评价此页

enable_fake_quant#

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

为模块启用伪量化。

如果适用,则为此模块启用伪量化。示例用法

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