Public Member Functions | |
def | __init__ (self, blobs, logging_frequency, i1=0, i2=0) |
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 get a certain entry from certain blobs. Args: blobs: list of blobs to get entry from logging_frequency: frequency for printing entry values to logs i1, i2: the first, second dimension of the blob. (currently, we assume the blobs to be 2-dimensional blobs). When i2 = -1, print all entries in blob[i1]
Definition at line 27 of file get_entry_from_blobs.py.