quantize_qat

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

执行量化感知训练并输出一个量化模型

参数

• model – 输入模型

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

• run_args – run_fn 的位置参数

返回

量化模型。