add_observer

class torch.quantization.add_observer_(module, qconfig_propagation_list=None, non_leaf_module_list=None, device=None, custom_module_class_mapping=None) [source]

Add observer for the leaf child of the module.

This function insert observer module to all leaf child module that has a valid qconfig attribute.

Parameters:

• module – input module with qconfig attributes for all the leaf modules that we want to quantize
• qconfig_propagation_list – a list of quantizable modules that will have observers added to them if they are leaf nodes
• device – parent device, if any
• non_leaf_module_list – list of non-leaf modules we want to add observer

Returns:

None, module is modified inplace with added observer modules and forward_hooks