使用snpe把onnx转成dlc的时候报错

weixin_32291749 2024-05-22 18:26:33

使用snpe把onnx转成dlc的时候报错:

Traceback (most recent call last):   File "/home/adil/snpe/snpe-1.61.0.3358/lib/python/qti/aisw/converters/onnx/onnx_to_ir.py", line 231, in convert     src_op.op_type)   File "/home/adil/snpe/snpe-1.61.0.3358/lib/python/qti/aisw/converters/common/converter_ir/translation.py", line 48, in apply_method_to_op     translation = self.__get_translation(op_type)   File "/home/adil/snpe/snpe-1.61.0.3358/lib/python/qti/aisw/converters/common/converter_ir/translation.py", line 35, in __get_translation     "Op is most likely not supported by the converter." % op_type)  KeyError: 'No translation registered for op type onnx_hardsigmoid. Op is most likely not supported by the converter.'  2023-10-19 11:28:15,854 - 209 - ERROR - Node /backbone/backbone/stem/conv/act/HardSigmoid: 'No translation registered for op type onnx_hardsigmoid. Op is most likely not supported by the converter.

...全文
1385 1 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
weixin_38498942 2024-05-23
  • 打赏
  • 举报
回复 1

您好,从log中分析,您使用的是SNPE-1.61。这个问题出在HardSigmoid这个算子本身是snpe-2.20才开始支持的,1.61还不支持。如果您知道模型中的什么位置使用了这个算子,您可以将其替换成其他激活函数,或者将其拆解为更基础的表示;如果您找不到您在哪里使用了HardSigmoid,那大概率是因为您使用了pytorch自带的HSwish算子。Pytorch自带的HSiwsh算子使用的是xHardSigmoid的表述,您可以自定义一个HSwish,将其改成xnn.functional.relu6(x+3, inplace=True)/6的表述。

3,298

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧