quantize_qat

class torch.ao.quantization.quantize_qat(model, run_fn, run_args, inplace=False)

執行量化感知訓練 (Quantization Aware Training) 並輸出一個量化模型

參數

• model – 輸入模型

• run_fn – 一個用於評估準備好的模型的函數，可以是一個簡單地運行準備好的模型的函數，或者是一個訓練迴圈

• run_args – run_fn 的位置參數

返回

量化模型。