在此一键设置昵称,即可参与社区互动!
newer
发帖: 11粉丝: 0
发消息 + 关注
求助,在使用ATC工具转换CRNN代码时,遇到以下报错:
在转换过程中,不断指定输出节点,最后定位到输入的双向LSTM层中,具体问题出现在下图中
在指定输出节点为红色的时候,转om模型成功,指定输出节点为浅蓝色的时候出现上面报错,模型转换失败,求指点。
分享文章到朋友圈
windyxyz
发帖: 49粉丝: 26
你好,推理场景下,不支持tf的while循环。while是框架的机制,推理上没有这个机制,不能直接用BasicLSTMCell。
您好,那推理场景中,支持双向LSTM吗,或者是该如何解决这问题,我在atlas800文档上,没看到该算子支持,感谢
caorp
发帖: 3粉丝: 7
麻烦楼主提供下模型,以及转换命令和相关的配置文件。
不好意思耽误太久时间了。
CRNN是支持的吗
评论 newer:这边有新环境,可能支持,所以要模型试试看。
评论 caorp:感谢,这边上传模型不成功,有大小限制,可以加你qq发您吗
小孬
发帖: 0粉丝: 0
您好,从计算图里面看出当前的LSTM使用的API函数是用小算子拼接出来的,并通过循环算子将单个时刻的计算串联起来,推理场景下,不支持tf的while循环。可行的方式是需要增加Scope融合,将这些小算子融合为统一的DynamicRNN算子,并拼接出双向的结构。
游客
回复主题
您发表的内容存在敏感词汇
如点击继续发布,文章中的敏感词将以*代替。
您好,那推理场景中,支持双向LSTM吗,或者是该如何解决这问题,我在atlas800文档上,没看到该算子支持,感谢
... 查看全部