专栏算法工具链LSS bev 添加depth监督qat loss不下降

LSS bev 添加depth监督qat loss不下降

已解决
张飞飞962024-03-04
114
3

用户您好,请详细描述您所遇到的问题,详细的描述有助于帮助我们快速定位,解决问题~Thanks♪(・ω・)ノ

1. 芯片型号:J5

2. 天工开物开发包 OpenExplorer 版本: J5_OE_1.1.37

3. 问题定位:LSS bev 添加depth监督qat loss不下降

4. 问题具体描述:在官方bev lss网络中添加depth监督,并依次对dataset/target/loss各部分都做了相应的修改,float训练loss可正常下降,float结果可视化也正常。但qat训练时depth_loss无法下降反而越训越大,qat训练完的推理效果较差。

想请教一下这种从网络中间模块引出一个输出来监督训练的,qat部分的修改还需要注意些什么?对于这个问题有没有什么问题排查方法或者建议,感谢~

下列为一些修改细节:

  1. depth pred,在这里为了方便计算loss,使用了DeQuantStub()

2.depth target


5. 如果提供的信息过多,建议进行打包并上传附件至百度网盘

算法工具链
征程5
评论3
0/1000
  • GOYU\XU
    Lv.2

    抱歉,这么久回答您的问题。请问qat训练是纯pytorch框架训练的,还是依赖地平线qat工具链进行的训练?

    2024-03-15
    1
    0
  • 张飞飞96
    Lv.1

    是基于地平线的工具链框架训练的

    2024-03-25
    0
    0
  • GOYU\XU
    Lv.2

    您好,请问方便提供训练源码及训练数据吗,方便这边问题复现及问题排查

    2024-03-26
    0
    0