专栏算法工具链swin transformer导出onnx报错

swin transformer导出onnx报错

已解决
默认383362025-04-17
51
3
您好，我在用oe包中qat模型示例下的export_onnx.py将 horizon_swin_transformer 的pth导出onnx过程中报错 floating points exception(core dumped) ，查询博客发现是torch1.13版本和地平线自定义算子冲突引入的bug，请问现在修复了吗？如果没有可否提供下该模型各个pth对应的onnx文件？之前博客下的网盘链接过期了。谢谢
算法工具链
征程5
评论2
0/1000
  • Huanghui
    Lv.5

    你好，你的docker镜像是什么版本的 。我来验证下

    2025-04-23
    0
    1
    • 默认38336回复Huanghui:

      Ubuntu_20_j5_gpu:v1.1.68

      2025-04-23
      0
  • Huanghui
    Lv.5

    不好意思，前端时间没来的及处理。经验证目前看确实存在该问题，无论是你说的v1.1.68还是新的v1.1.77，已经反馈给研发，请关注发版情况！

    2025-06-02
    0
    0