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) |
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.