专栏算法工具链vision tranformer模型在J3的部署

vision tranformer模型在J3的部署

已解决
kkkkk2024-09-09
62
3

用户您好，请详细描述您所遇到的问题，详细的描述有助于帮助我们快速定位，解决问题~Thanks♪(･ω･)ﾉ

1. 芯片型号：J3

2. 天工开物开发包 OpenExplorer 版本： J3_OE_1.16.2

3. 问题定位：模型转换、板端部署

4. 问题具体描述：vision transformer模型有一半算子仅J3cpu支持，请问J3有无针对vision transformer模型转换、板端部署的优化，若有能否提供相应优化方法？

算法工具链
征程3
评论2
0/1000
  • Huanghui
    Lv.5

    你好，这个问题比较大，请check算子在cpu上执行的原因，并做下面的尝试。

    1. 算子确实无法再BPU上支持的：根据模型设计，确定是否可以将不支持的算子替换成支持的算子。

    2. 算子在符合条件是可以支持：调整算子的数据到BPU可支持的范围。


    2024-09-09
    0
    1
    • kkkkk回复Huanghui:

      感谢回复，我尝试下。

      2024-09-10
      0
  • yy1234
    Lv.2

    这玩意要是能做出来也太牛了

    2024-09-10
    0
    0