enable_fake_quant
- class torch.ao.quantization.fake_quantize.enable_fake_quant(mod)[source]
Enable fake quantization for this module, if applicable. Example usage:
# model is any PyTorch model model.apply(torch.ao.quantization.enable_fake_quant)
Enable fake quantization for this module, if applicable. Example usage:
# model is any PyTorch model
model.apply(torch.ao.quantization.enable_fake_quant)