Public Member Functions | |
def | __init__ (self, blobs, logging_frequency) |
def | modify_net (self, net, init_net=None, grad_map=None, blob_to_device=None, modify_output_record=False) |
def | field_name_suffix (self) |
Public Member Functions inherited from caffe2.python.modeling.net_modifier.NetModifier | |
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) |
This class modifies the net passed in by adding ops to compute statistics for certain blobs. For each blob in the list, its min, max, mean and standard deviation will be computed. Args: blobs: list of blobs to compute norm for logging_frequency: frequency for printing norms to logs
Definition at line 12 of file compute_statistics_for_blobs.py.