model_helper Namespace Reference

Module caffe2.python.model_helper. More...

Classes
class	ModelHelperBase
class	ParameterInfo
class	ParameterType

Functions
def	ExtractPredictorNet (net_proto, input_blobs, output_blobs, device=None, renames=None, disabled_inputs=None)

Detailed Description

Module caffe2.python.model_helper.

Function Documentation

ExtractPredictorNet()

def model_helper.ExtractPredictorNet	(		net_proto,
			input_blobs,
			output_blobs,
			device = None,
			renames = None,
			disabled_inputs = None
	)

Takes a model net for training and returns a net which can be
used for prediction. For example, all gradient operators and
input operators are removed.
@param net_proto protobuf of the net you want to process (net.Proto())
@param input_blobs list/set of blob names that are the inputs of predictor
@param output_blobs list/set of blob names that are outputs of predictor
@param device optional device option that is assigned
@param renames dictionary of blob name to a new name (optional)
@param disabled_inputs optional set of blobs that are 'switched off'. This
            will cause branches with those blobs as inputs to be removed

Definition at line 355 of file model_helper.py.