Shortcuts

获取默认的qconfig映射

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

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

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

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

Return type

QConfigMapping

优云智算