Shortcuts

获取默认的QAT量化配置映射

class torch.ao.quantization.qconfig_mapping.get_default_qat_qconfig_mapping(backend='x86', version=1)[源代码]

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

Parameters
  • backend (*) – 默认qconfig映射的量化后端,应为以下之一 [“x86” (默认), “fbgemm”, “qnnpack”, “onednn”]

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

Return type

QConfigMapping

优云智算