专栏算法工具链关于浮点模型训练

关于浮点模型训练

已解决
Lmh2024-09-27
95
3

J5 J5_OE_1.1.74

问题三、关于浮点模型训练

      我们在calibration阶段时往模型代码里加入了一些地平线的修饰函数 from horizon_plugin_pytorch.utis.fx_helper import wrap 。然后将 model经过prepare_qat_fx函数后获得的calib_model以及model本身都没办法进行原来的浮点模型训练,训练model报错内容是:经过fx wrap修饰器修饰后的fx_dim函数参数发生了变化,1调用fx_dim(img)时报错多传入了一个参数,2调用fx_dim()时发现好像传入了默认参数BEVFormer;训练qat_model报错内容是:list不能deep copy

请问在加了修饰函数之后的模型就不再能执行浮点训练了吗?

 

1、Calib_model的报错

2、float_model的报错

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

    问题已经收到

    2024-09-27
    0
    0
  • Huanghui
    Lv.5
    你好,从目前了解的信息看,经过fx wrap修饰器修饰的函数是不影响浮点训练的。用 fx.wrap 修饰后的函数会作为整体运行在CPU上,需要保证input 是浮点数据。
    2024-10-24
    0
    0
  • Huanghui
    Lv.5

    客户您好,长时间未收到你的答复,相信问题已解。如对此尚存疑问欢迎新帖讨论,感谢您的参与!

    2024-10-29
    0
    0