[西工大]加入lstm模块的模型，pb转om的过程中，lstmcell处出现错误

已经通过在线推理验证pb模型正确性
之前使用无lstm的模型，已经能够成功完成在Ascend上的推理

具体报错信息如下
[ERROR] GE(11851,atc):2021-01-16-18:13:04.811.757 [framework/domi/parser/tensorflow/tensorflow_parser.cc:460]11851 CheckoutInputNum: ErrorNo: -1(failed) op [W_0/actor/rnn/while/lstm_cell], type[BasicLSTMCell], The input number of tensorflow[3] should be equal to factory size[5]
[ERROR] GE(11851,atc):2021-01-16-18:13:04.820.961 [framework/domi/parser/tensorflow/tensorflow_parser.cc:1271]11851 ParseAllGraph: ErrorNo: -1(failed) Add op[W_0/actor/rnn/while/lstm_cell] failed
[ERROR] GE(11851,atc):2021-01-16-18:13:04.823.161 [framework/domi/parser/tensorflow/tensorflow_parser.cc:1141]11851 Parse: ErrorNo: -1(failed) Failed to parse graph root, instance name alien_model
[ERROR] GE(11851,atc):2021-01-16-18:13:04.828.688 [framework/domi/parser/tensorflow/tensorflow_parser.cc:1073]11851 Parse: ErrorNo: -1(failed) Parser graph alien_model failed.
[ERROR] GE(11851,atc):2021-01-16-18:13:04.830.709 [framework/domi/offline/../session/omg.cc:671]11851 ParseGraph: ErrorNo: -1(failed) ATC model parse ret fail. Error Code:0xFFFFFFFF(Failed)
[ERROR] GE(11851,atc):2021-01-16-18:13:04.843.086 [framework/domi/offline/main.cc:900]11851 GenerateModel: ErrorNo: -1(failed) ATC Parse graph domi::FAILED
[ERROR] GE(11851,atc):2021-01-16-18:13:04.846.179 [framework/domi/offline/main.cc:901]11851 GenerateModel: ErrorNo: -1(failed) ATC Generate execute failed
模型文件连接：https://gitee.com/HengYueTM/a3-c/blob/master/Infer/model/alien_model.pb
希望能够帮忙分析下具体问题所在

Ascend/modelzoo
暂停

内容风险标识

评论 (41)

Ascend/modelzoo暂停 .gitee-modal { width: 500px !important; }

内容风险标识