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
您好,地平线工具链在持续迭代优化,为了给您提供更好的服务,希望您能抽出3分钟左右的时间,将您在使用工具链期间的感受和建议告诉我们,您的宝贵意见对我们很重要,非常感谢!问卷链接:https://wenjuan.feishu.cn/m/cfm?t=st64p6krU3Ji-yvhv
颜值即正义
3
你好,在J5_OE_1.1.57的docker中已经复现此报错,经过分析后,报错原因为torch 1.13.0引入的bug, 请您参考以下文档尝试一下QAT部署通路https://developer.horizon.ai/forumDetail/143772473308124166

谢谢快速反馈!简单看了下你上面提供的参考文档,仍然没办法导出onnx模型对吧,如果想导出onnx模型,是否还有其它的办法?我之前有将原生的swin-transformer模型导出为onnx模型,是可以正常导出的。
颜值即正义
5
你好,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天,请尽快下载