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.