Rate this Page

get_observer_state_dict#

class torch.ao.quantization.observer.get_observer_state_dict(mod)[source]#

Returns the state dict corresponding to the observer stats. Traverse the model state_dict and extract out the stats.