megengine.quantization.quantize

quantize(module, inplace=True, mapping=None)

使用 apply 将 QATModule 递归地转换为 QuantizedModule 。

参数

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

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

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

上一页

megengine.quantization.quantize_qat

下一页

megengine.quantization.apply_easy_quant