建议使用以下浏览器,以获得最佳体验。 IE 9.0+以上版本 Chrome 31+ 谷歌浏览器 Firefox 30+ 火狐浏览器
请选择 进入手机版 | 继续访问电脑版
设置昵称

在此一键设置昵称,即可参与社区互动!

确定
我再想想
选择版块
标签
您还可以添加5个标签
  • 没有搜索到和“关键字”相关的标签
  • 云产品
  • 解决方案
  • 技术领域
  • 通用技术
  • 平台功能
取消

捕头不捕头

发帖: 3粉丝: 0

级别 : 新手上路

发消息 + 关注

发表于2020年02月27日 09:41:55 601 4
直达本楼层的链接
楼主
显示全部楼层
[问题求助] 自定义算子Reduction在转换模型时报错

按文档自定义算子层操作,算子也编译成功,但在转换模型的时候报错,部分日志如下:

log.jpg

举报
分享

分享文章到朋友圈

分享文章到微博

捕头不捕头

发帖: 3粉丝: 0

级别 : 新手上路

发消息 + 关注

发表于2020年02月27日 14:10:35
直达本楼层的链接
沙发
显示全部楼层

呼叫大神,紧急啊

点赞 评论 引用 举报

MrRoot

发帖: 10粉丝: 14

级别 : 版主

发消息 + 关注

发表于2020年02月27日 14:56:09
直达本楼层的链接
板凳
显示全部楼层

根据报错,入参的轴的数据类型必须是整数,检查下你的reduction函数接收到的axis的值是否为整数呢?

评论
捕头不捕头 2020-2-27 15:04 评论

请问这个怎么检查,这个参数在哪里传入的,是prototxt文件吗

... 查看全部
捕头不捕头 2020-2-27 15:53 评论

prototxt里面的axis值真的是整数,整的不能再整的整数

... 查看全部
点赞 评论 引用 举报

捕头不捕头

发帖: 3粉丝: 0

级别 : 新手上路

发消息 + 关注

发表于2020年03月02日 16:55:19
直达本楼层的链接
地板
显示全部楼层

回复:MrRoot 发表于 2020-2-27 14:56 根据报错,入参的轴的数据类型必须是整数,检查下你的reduction函数接收到的axis的值是否为整数呢?

自定义tile层在转换的时候又出问题了,您说的参数顺序我加进去试了试,都还是报错,报错日志部分如下:

[INFO] FMK:2020-03-02-16:25:23.248.666 SetInputDesc:framework/domi/omg/../omg/model/model_builder.cpp:504:"dims[0]: 1"

[INFO] FMK:2020-03-02-16:25:23.248.674 SetInputDesc:framework/domi/omg/../omg/model/model_builder.cpp:504:"dims[1]: 256"

[INFO] FMK:2020-03-02-16:25:23.248.681 SetInputDesc:framework/domi/omg/../omg/model/model_builder.cpp:504:"dims[2]: 1"

[INFO] FMK:2020-03-02-16:25:23.248.689 SetInputDesc:framework/domi/omg/../omg/model/model_builder.cpp:504:"dims[3]: 1"

[ERROR] FMK:2020-03-02-16:25:23.248.769 AdjustParams:framework/domi/omg/../omg/model/op_builder/tile_op_builder.cpp:45:"Weight num of Tile Op must be 1."

[ERROR] FMK:2020-03-02-16:25:23.248.779 Build:framework/domi/omg/../omg/model/op_builder/op_builder.cpp:32:""Adjust params" failed. Node: tile1."

[ERROR] FMK:2020-03-02-16:25:23.248.785 SetInputOutputDesc:framework/domi/omg/../omg/model/model_builder.cpp:575:"Op build failed. Node: tile1."

[ERROR] FMK:2020-03-02-16:25:23.248.800 Build:framework/domi/omg/../omg/model/model_builder.cpp:2838:"SetInputOutputDesc Failed!"

[ERROR] FMK:2020-03-02-16:25:23.248.806 Generate:framework/domi/omg/omg.cpp:813:"OMG builder Build() return fail."

[ERROR] FMK:2020-03-02-16:25:23.255.824 main:framework/domi/omg_main/main.cpp:791:"OMG Generate execute failed!!"

OMG generate offline model failed. Please see the log or pre-checking report for more details.

[INFO] RUNTIME:2020-03-02-16:25:23.257.859 11876 runtime/feature/src/driver.cc:57 ~Driver:deconstruct driver


评论
MrRoot 2020-3-2 18:02 评论

是换了个模型换了个算子吗?

... 查看全部
捕头不捕头 2020-3-3 09:02 评论

对,模型和算子都换了,自定义了各tile层,然后还有这个问题,还望不吝赐教。非常紧急,目前就差这个算子了

... 查看全部
捕头不捕头 2020-3-3 09:04 评论

评论 MrRoot:对,模型和算子都换了,自定义了各tile层,然后还有这个问题,还望不吝赐教。非常紧急,目前就差这个算子了

... 查看全部
点赞 评论 引用 举报

捕头不捕头

发帖: 3粉丝: 0

级别 : 新手上路

发消息 + 关注

发表于2020年03月03日 10:04:32
直达本楼层的链接
5#
显示全部楼层

有人吗

评论
MrRoot 2020-3-3 18:06 评论

你的模型和代码再发我看下呢?

... 查看全部
捕头不捕头 2020-3-4 14:12 评论

评论 MrRoot:模型已邮件发送,还望帮忙查看下,谢谢

... 查看全部
点赞 评论 引用 举报

游客

富文本
Markdown
您需要登录后才可以回帖 登录 | 立即注册