評價此頁

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