专栏算法工具链horizon swin-t onnx模型转换报错

horizon swin-t onnx模型转换报错

已解决
luwenjuan902023-07-17
48
4
1.芯片型号：J5
2.天工开物开发包OpenExplorer版本：J5_OE_1.1.57
3.问题定位：模型转换
4.问题具体描述：使用tools中的脚本export_onnx.py将horizon_swin_transformer转换成onnx模型时，报错了。执行命令：python3 tools/export_onnx.py -c config/classification/horizon_swin_transformer.py。报的错误为：floating points exception(core dumped)
算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2
    你好，在J5_OE_1.1.57的docker中已经复现此报错，经过分析后，报错原因为torch 1.13.0引入的bug, 请您参考以下文档尝试一下QAT部署通路https://developer.horizon.ai/forumDetail/143772473308124166
    2023-07-17
    0
    2
    • luwenjuan90回复颜值即正义:

      谢谢快速反馈！简单看了下你上面提供的参考文档，仍然没办法导出onnx模型对吧，如果想导出onnx模型，是否还有其它的办法？我之前有将原生的swin-transformer模型导出为onnx模型，是可以正常导出的。

      2023-07-18
      0
    • 颜值即正义回复luwenjuan90:
      你好，tools文件夹下的export_onnx.py脚本导出所有模型的，在1.10的torch里是可以导出的，但是由于torch 1.13有bug导致swin-t中的某个算子无法导出了，而且暂时是无法修复的。
      所以我们在网盘上提供了在torch1.10.0下导出的swint的onnx模型
      下载链接：https://pan.horizon.ai/index.php/s/yCzHnQkGNEJXC2G

      有效期为6天，请尽快下载

      2023-07-19
      0
  • 颜值即正义
    Lv.2
    您好，地平线工具链在持续迭代优化，为了给您提供更好的服务，希望您能抽出3分钟左右的时间，将您在使用工具链期间的感受和建议告诉我们，您的宝贵意见对我们很重要，非常感谢！问卷链接：https://wenjuan.feishu.cn/m/cfm?t=st64p6krU3Ji-yvhv
    2023-07-17
    0
    0