Rate this Page

load_observer_state_dict#

class torch.ao.quantization.observer.load_observer_state_dict(mod, obs_dict)[source]#

Given input model and a state_dict containing model observer stats, load the stats back into the model. The observer state_dict can be saved using torch.ao.quantization.get_observer_state_dict