启用观察者

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

开启此模块的观察功能。

如果适用,请为此模块启用观察功能。示例如下:

# model is any PyTorch model
model.apply(torch.ao.quantization.enable_observer)
本页目录