get_observer_state_dict
-
class
torch.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.
torch.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.