mindspore_gs.ptq.PTQConfig

查看源文件
class mindspore_gs.ptq.PTQConfig(mode=PTQMode.QUANTIZE, backend=BackendTarget.NONE)[源代码]

用于配置 MindSpore Golden Stick 后量化算法的数据类。

参数:
  • mode (mindspore_gs.ptq.PTQMode) - 用于配置 PTQ 算法的模式, QUANTIZE 表示为量化阶段, DEPLOY 表示为部署阶段。

  • backend (mindspore_gs.common.BackendTarget) - 用于配置 PTQ 算法转换为真实量化网络后端, NONE 表示无具体后端,为通用网络.``ASCEND`` 表示为昇腾后端,会在网络中插入昇腾相关算子。

异常:
  • ValueError - mode 输入不在 [PTQMode.QUANTIZE, PTQMode.DEPLOY] 中。

  • ValueError - backend 输入不在 [BackendTarget.NONE, BackendTarget.ASCEND] 中。