Shortcuts

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources