快捷方式

prepare_qat

torch.ao.quantization.prepare_qat(model, mapping=None, inplace=False)[源][源]

為量化校準或量化感知訓練 (QAT) 準備模型的副本,並將其轉換為量化版本。

量化配置應預先分配給各個子模組的 .qconfig 屬性。

引數
  • model – 要進行就地修改的輸入模型

  • mapping – 對映浮點模組到要替換的量化模組的字典。

  • inplace – 就地執行模型轉換,原模組會被修改

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源