評價此頁

enable_fake_quant#

class torch.ao.quantization.fake_quantize.enable_fake_quant(mod)[原始碼]#

為模組啟用偽量化。

如果適用,為該模組啟用偽量化。示例用法

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