专栏算法工具链【求】地平线 swin transformer的onnx模型，哪个大哥有啊

【求】地平线 swin transformer的onnx模型，哪个大哥有啊

已解决
珠穆朗玛峰修电梯2024-12-16
100
10

【求】地平线 swin transformer的onnx模型，哪个大哥有啊

地平线OE包里没有 swin transformer的onnx模型，大家有用吗
算法工具链
征程5征程6社区活动技术深度解析
+1
评论4
0/1000
  • Jade-self
    Lv.5
    地平线的swin transformer是参考算法，源码、配置文件以及对应的每个阶段权重均已提供。
    源码在docker中。
    配置文件可见：OE/samples/ai_toolchain/horizon_model_train_sample/scripts/configs/classification/horizon_swin_transformer_imagenet.py
    权重可见：OE/samples/ai_toolchain/horizon_model_train_sample/scripts/configs/classification/README.md
    2024-12-16
    1
    3
    • 珠穆朗玛峰修电梯回复Jade-self:
      老哥有onnx文件吗beishang_org.svg
      2024-12-17
      0
    • Jade-self回复珠穆朗玛峰修电梯:

      没有，建议学习一下qat

      2024-12-17
      1
    • 珠穆朗玛峰修电梯回复Jade-self:

      嗯呢嗯呢

      2024-12-17
      0
  • Huanghui
    Lv.5

    J5还是J6？需求描述清楚一下

    2024-12-16
    0
    1
    • 珠穆朗玛峰修电梯回复Huanghui:

      j6的

      2024-12-16
      0
  • Huanghui
    Lv.5

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

    2025-01-09
    0
    0
  • DR_KAN
    Lv.4

    这个属于参考算法，是以源码形式提供的，并且专属于QAT，需要自行使用QAT方法训练量化编译，OE包有直接提供hbm

    2024-12-17
    0
    2
    • 珠穆朗玛峰修电梯回复DR_KAN:

      用ptq的量化方法可以转换swint吗

      2024-12-17
      0
    • gaoh回复珠穆朗玛峰修电梯:

      ptq可以量化swint，这里建议导出参考算法结构的swint onnx，传入config文件并使用samples/ai_toolchain/horizon_model_train_sample/scripts/tools/export_onnx.py脚本可以导出onnx模型

      2024-12-17
      0