quantize_qat ¬

class torch.ao.quantization.quantize_qat(model, run_fn, run_args, inplace=False)[source][source] ¬

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

参数:

• model – 输入模型

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

• run_args – run_fn 的位置参数

返回值:

量化模型。