/usr/local/lib64/python3.6/site-packages/caffe2/python/modeling
NameSizeModeActions
__pycache__/-0755rm
compute_histogram_for_blobs.py35080644editdlrm
compute_histogram_for_blobs_test.py49470644editdlrm
compute_norm_for_blobs.py36420644editdlrm
compute_norm_for_blobs_test.py79420644editdlrm
compute_statistics_for_blobs.py19090644editdlrm
compute_statistics_for_blobs_test.py28700644editdlrm
get_entry_from_blobs.py31300644editdlrm
get_entry_from_blobs_test.py35380644editdlrm
gradient_clipping.py57660644editdlrm
gradient_clipping_test.py100160644editdlrm
initializers.py53780644editdlrm
initializers_test.py21060644editdlrm
net_modifier.py8230644editdlrm
parameter_info.py14380644editdlrm
parameter_sharing.py38480644editdlrm
parameter_sharing_test.py72230644editdlrm
__init__.py00644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/modeling/net_modifier.py (823B)
import abc class NetModifier(metaclass=abc.ABCMeta): """ 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) """ def __init__(self): pass @abc.abstractmethod def modify_net(self, net, init_net=None, grad_map=None, blob_to_device=None): pass def __call__(self, net, init_net=None, grad_map=None, blob_to_device=None, modify_output_record=False): self.modify_net( net, init_net=init_net, grad_map=grad_map, blob_to_device=blob_to_device, modify_output_record=modify_output_record)