专栏算法工具链qat时候，model.train未生效解决

qat时候，model.train未生效解决

李俊何?2025-07-11
35
6
MARK
在进行qat训练之前需要将model设置为train(),  但是会遇到设置了之后依然会报错未调用model.train的问题，此时可能是一些子模块未被设置，
此时需要遍历这个模型的所有子模块
算法工具链
技术深度解析征程6
评论2
0/1000
  • Huanghui
    Lv.5

    你好，是不是伪量化节点状态没设置对？

    2025-07-11
    0
    0
  • Huanghui
    Lv.5

    你先参考下这段QAT训练的代码

    2025-07-11
    0
    4
    • 李俊何?回复Huanghui:

      大佬周五好，我是严格按照这个代码来的，但是确实遇到了这个问题，这是我的代码流程

      2025-07-11
      0
    • Huanghui回复李俊何?:

      你刚刚遍历每一个模块，逐层设置train()是没有报错吗？

      2025-07-11
      0
    • 李俊何?回复Huanghui:
      是的 我这样做了之后就能正常进行下去了xiaoku_org.svg，我等会儿会看一下我的精度是否又提升，检查一下是不是我的模型prapare的有问题（大概率不会，我对calib之后的模型进行可视化是正常的，精度掉了8个点也正常，可视化正常）
      2025-07-11
      0
    • Huanghui回复李俊何?:

      可以 ，流程能顺利进行下去就行。

      2025-07-11
      1