caffe2.python.modeling.net_modifier.NetModifier Class Reference

Inheritance diagram for caffe2.python.modeling.net_modifier.NetModifier:

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, modify_output_record=False)

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)
    modifier(net)

Definition at line 10 of file net_modifier.py.

---

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

• caffe2/python/modeling/net_modifier.py