lstm atc模型转换
收藏回复举报
lstm atc模型转换
t('forum.solved') 已解决
发表于2024-07-26 17:10:35
0 查看

我使用tensorflow2.6.5训练了一个lstm网络模型,保存为pb格式,用atc工具转换成om格式时报输入形状不合法,lstm网络的输入形状为(151,4)已经在input_shape中声明了,input_format也改成的ND,

求助一下为什么会报错

image.png

image.png

具体日志如下:

我要发帖子