enable_fake_quant

class torch.ao.quantization.fake_quantize.enable_fake_quant(mod)

Enable fake quantization for the module.

Enable fake quantization for this module, if applicable. Example usage:

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