专栏算法工具链QAT量化流程

QAT量化流程

已解决
默认328322024-07-19
82
4

J5 J5_OE_1.1.74 算法工具链 7.3.2获取浮点模型，中提到可以直接加载训练好的模型，请问有加载示例吗？加载后如何插入QuantStub和DequantStub？

算法工具链
技术深度解析征程5
评论2
0/1000
  • kotei左文亮
    Lv.3

    浮点模型训练之前是要先进行改造，可查看样例：


    2024-07-19
    0
    1
    • 默认32832回复kotei左文亮:

      可能我表达得不是很清楚，算法工具链里面的直接加载也是通过你给的这个示例的方法加载吗？（图片最后一句若有已训练好的模型直接加载即可）

      2024-07-21
      0
  • kotei左文亮
    Lv.3

    上面是mobv2的样例，网络模型换成自己的就可加载了。

    2024-07-22
    0
    1
    • 默认32832回复kotei左文亮:
      好的，谢谢你的解答，但是训练好的模型加载是通过torch.load直接进行加载的如何插入QuantStub和DequantStub？
      2024-07-22
      0