专栏算法工具链J6E QAT模型训练异常,prepare模型报错

J6E QAT模型训练异常,prepare模型报错

已解决
默认636802025-04-17
45
1

J6E模型QAT训练的时候报错,模型包含backbone,fpn, head多模块,多任务模型,在准备QAT模型的时候prepare报错:

WARNING: check model failed. The common reason is that inputs required for multiple forward are inconsistent. solution:

1. call horizon_plugin_pytorch.utils.check_model.check_qat_model individually

2. rewrite model forward to ensure that inputs of multiple forward are consistent.

 

(2)输出的信息如下, 是否需要修改?

INFO: cat_19 doesn't support high precisionoutput.

 

(3)模型开始训练,报错

TypeError: rand() received an invalid combination of arguments - got (list, device=torch.device, dtype=QuantDType), but expected one of:

 

 

错误如附件,还请帮忙看看, 谢谢!

附件:
算法工具链
征程6
评论1
0/1000
  • Huanghui
    Lv.5

    工具链目前已发布新的版本V3.7.0,推荐你使用新版本进行测试,如有问题可以通过新的帖子反馈我们,我们会及时跟踪处理,感谢您的理解与支持~

    2026-03-03
    0
    0