megengine.quantization.apply_easy_quant

apply_easy_quant(module, data, start=0.8, stop=1.2, num=40)

EasyQuant 的实现，搜索最合适的scale。

参数

• module (Module) – 根模块。

• data (Tensor) – 用于搜索scale的输入。

• start (float) – 搜索下界。

• stop (float) – 搜索上界。

• num (int) – 搜索区间的采样数量。

• module – Module:

上一页

megengine.quantization.quantize

下一页

megengine.quantization.enable_fake_quant