专栏算法工具链hbm编译问题

hbm编译问题

白菜飘落2023-08-04
43
2
1.芯片型号：J5
2.天工开物开发包OpenExplorer版本：horizon_j5_open_explorer_v1.1.29_20221103
3.问题定位：编译hbm
4.问题具体描述：当前我已经有一个编译好的hbm。模型的最后一层是argmax，现在我需要将argmax从hbm里去掉。我能直接在模型的forward里去掉argmax，然后直接编译hbm吗？
算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    可以啊，需要注意的是，模型改了，编译后hbm的输出也会跟着改，建议你尝试一下，观察改动前后输出的变化是否符合预期，毕竟操作起来也不复杂~

    此外，注意到你使用的OE版本比较久远了，如果是用于开发的话，工具链持续迭代更新，欢迎更新到当前最新版本OE1.1.60，获取链接：https://developer.horizon.cc/forumDetail/118363912788935318

    2023-08-04
    0
    0
  • 颜值即正义
    Lv.2

    您好，地平线工具链在持续迭代优化，为了给您提供更好的服务，希望您能抽出3分钟左右的时间，将您在使用工具链期间的感受和建议告诉我们，您的宝贵意见对我们很重要，非常感谢！

    2023-08-04
    0
    0