Shortcuts

量化_qat

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

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

Parameters
  • 模型 – 输入模型

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

  • run_argsrun_fn 的位置参数

Returns

量化模型。

优云智算