prepare_qat¶ 類 torch.ao.quantization.prepare_qat(model, mapping=None, inplace=False)[源][源]¶ 為量化校準或量化感知訓練 (QAT) 準備模型的副本,並將其轉換為量化版本。 量化配置應預先分配給各個子模組的 .qconfig 屬性。 引數 model – 要進行就地修改的輸入模型 mapping – 對映浮點模組到要替換的量化模組的字典。 inplace – 就地執行模型轉換,原模組會被修改