pytorch / 2 / generated / torch.ao.quantization.fake_quantize.disable_fake_quant.html

disable_fake_quant

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

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

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.ao.quantization.fake_quantize.disable_fake_quant.html