quantize_qat

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

執行量化感知訓練並輸出一個量化模型

引數

• model – 輸入模型

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

• run_args – run_fn 的位置引數

返回

量化模型。