quantize_qat

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

进行量化感知训练并输出量化模型

参数

• model – 输入模型

• run_fn – 用于评估准备好的模型的函数，可以是简单地运行准备好的模型的函数，也可以是训练循环

• run_args – run_fn 的位置参数

返回

量化模型。