快捷方式

quantize_qat

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

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

参数
  • model – 输入模型

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

  • run_argsrun_fn 的位置参数

返回值

量化模型。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源