Shortcuts

禁用伪量化

class torch.ao.quantization.fake_quantize.disable_fake_quant(mod)[源代码]

禁用模块的伪量化。

如果适用,禁用此模块的伪量化。示例用法:

# model 是任何 PyTorch 模型
model.apply(torch.ao.quantization.disable_fake_quant)
优云智算