量化

class torch.ao.quantization.quantize(model, run_fn, run_args, mapping=None, inplace=False)

使用訓練後靜態量化來量化輸入的浮點模型。

首先，它會準備模型以進行校準，然後呼叫 run_fn，這將執行校準步驟，之後我們會將模型轉換為量化模型。

參數

• model – 輸入的浮點模型

• run_fn – 用於校準已準備模型的校準函數

• run_args – run_fn 的位置參數

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

• mapping – 原始模組類型和量化對應物之間的對應關係

返回值

量化模型。