quantize

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 – 原始模組型別與量化對應物之間的對應關係

返回

量化模型。