Rate this Page

enable_observer#

class torch.ao.quantization.fake_quantize.enable_observer(mod)[source]#

Enable observation for this module.

Enable observation for this module, if applicable. Example usage:

# model is any PyTorch model
model.apply(torch.ao.quantization.enable_observer)