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: