启用假量化

classtorch.ao.quantization.fake_quantize.enable_fake_quant(mod)[源代码]

为模块开启假量化。

如果适用，为该模块启用伪量化。示例用法：

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