Public Member Functions | |
def | __init__ (self, blobs, logging_frequency, num_buckets=30, lower_bound=0.0, upper_bound=1.0, accumulate=False) |
def | modify_net (self, net, init_net=None, grad_map=None, blob_to_device=None, modify_output_record=False) |
def | field_name_suffix (self) |
![]() | |
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 histogram for certain blobs. Args: blobs: list of blobs to compute histogram for logging_frequency: frequency for printing lower_bound: left boundary of histogram values upper_bound: right boundary of histogram values num_buckets: number of buckets to use in [lower_bound, upper_bound) accumulate: boolean to output accumulate or per-batch histogram
Definition at line 12 of file compute_histogram_for_blobs.py.