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.
Returns the state dict corresponding to the observer stats. Traverse the model state_dict and extract out the stats.