megengine.quantization.quantize

quantize(module, inplace=True, mapping=None)[源代码]

使用 applyQATModule 递归地转换为 QuantizedModule

参数
  • module (Module) – 根模块,以此开始递归地进行转换。

  • inplace (bool) – 是否对子模块进行原地转换。

  • mapping (Optional[dict]) – 一个用来指示如何把 QATModule 转换为 QuantizedModule 的字典。将会和内部的默认的转换映射字典合并。