快捷方式

enable_fake_quant

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

为模块启用伪量化。

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

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

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源