加载观察者状态字典

函数torch.ao.quantization.observer.load_observer_state_dict(mod, obs_dict)[源代码]

给定输入模型和一个包含模型观察统计信息的 state_dict,将这些统计信息重新载入到模型中。可以使用 torch.ao.quantization.get_observer_state_dict 来保存观察器的状态字典。

本页目录