评价此页

get_default_qat_qconfig_mapping#

class torch.ao.quantization.qconfig_mapping.get_default_qat_qconfig_mapping(backend='x86', version=1)[source]#

返回量化感知训练的默认 QConfigMapping。

参数
  • backend (*) – 默认 qconfig 映射的量化后端,应为 [“x86” (默认), “fbgemm”, “qnnpack”, “onednn”] 中的一个

  • version (*) – 默认 qconfig 映射的版本

返回类型

QConfigMapping