Shortcuts

启用观察者

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

启用此模块的观察功能。

如果适用,启用此模块的观察功能。示例用法:

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