量化QAT

classtorch.ao.quantization.quantize_qat(model, run_fn, run_args, inplace=False)[源代码]

进行量化感知训练，并生成量化模型

参数

• 模型 – 输入模型

• run_fn — 用于评估准备好的模型的函数，它可以是直接运行模型的函数，也可以是训练循环。

• run_args - 传递给run_fn的位置参数

返回值

量化模型。