Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.modeling.net_modifier.NetModifier Class Reference
Inheritance diagram for caffe2.python.modeling.net_modifier.NetModifier:
caffe2.python.modeling.compute_histogram_for_blobs.ComputeHistogramForBlobs caffe2.python.modeling.compute_norm_for_blobs.ComputeNormForBlobs caffe2.python.modeling.compute_statistics_for_blobs.ComputeStatisticsForBlobs caffe2.python.modeling.gradient_clipping.GradientClipping

Public Member Functions

def __init__ (self)
def modify_net (self, net, init_net=None, grad_map=None, blob_to_device=None)
def __call__ (self, net, init_net=None, grad_map=None, blob_to_device=None)

Detailed Description

An abstraction class for supporting modifying a generated net.
Inherited classes should implement the modify_net method where
related operators are added to the net.

Example usage:
    modifier = SomeNetModifier(opts)

Definition at line 10 of file

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