get_observer_dict

class torch.quantization.get_observer_dict(mod, target_dict, prefix='') [source]

Traverse the modules and save all observers into dict. This is mainly used for quantization accuracy debug :param mod: the top module we want to save all observers :param prefix: the prefix for the current module :param target_dict: the dictionary used to save all the observers