caffe2.python.layers.build_index.MapToRange Class Reference

Inheritance diagram for caffe2.python.layers.build_index.MapToRange:

Public Member Functions
def	__init__ (self, model, input_record, max_index, name='map_to_range', kwargs)
def	add_train_ops (self, net)
def	add_eval_ops (self, net)
def	add_ops (self, net)
Public Member Functions inherited from caffe2.python.layers.layers.ModelLayer
def	__init__ (self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs)
def	get_type (self)
def	predict_input_record (self)
def	input_record (self)
def	predict_output_schema (self)
def	predict_output_schema (self, output_schema)
def	output_schema (self)
def	output_schema (self, output_schema)
def	get_parameters (self)
def	get_fp16_compatible_parameters (self)
def	get_memory_usage (self)
def	add_init_params (self, init_net)
def	create_param (self, param_name, shape, initializer, optimizer, ps_param=None, regularizer=None)
def	get_next_blob_reference (self, name)
def	add_operators (self, net, init_net=None, context=InstantiationContext.TRAINING)
def	add_ops (self, net)
def	add_eval_ops (self, net)
def	add_train_ops (self, net)
def	add_ops_to_accumulate_pred (self, net)
def	add_param_copy_operators (self, net)
def	export_output_for_metrics (self)
def	export_params_for_metrics (self)

Public Attributes
	max_index
	handler
	output_schema
Public Attributes inherited from caffe2.python.layers.layers.ModelLayer
	name
	model
	kwargs
	request_only
	precomputation_request_only
	precomputation_object_only
	eval_output_schema
	tags
	params

Detailed Description

This layer aims to build a mapping from raw keys to indices within [0, max_index).
The mapping is continuously built during training. The mapping will be frozen during
evaluation and prediction. Unseen keys will be assigned to index 0.

Definition at line 12 of file build_index.py.

---

The documentation for this class was generated from the following file:

• caffe2/python/layers/build_index.py