获取默认的量化感知训练 QConfigMapping ¶

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

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

参数:

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

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

返回类型:

QConfigMapping