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

@张韦全 使用新版本atc后,仍然有报错,日志如下:

Start to convert model

export PATH=/usr/local/python3.7.5/bin:$PATH:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/ccec_compiler/bin:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/bin && export PYTHONPATH=$PYTHONPATH:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/python/site-packages:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/python/site-packages/auto_tune.egg/auto_tune:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/python/site-packages/schedule_search.egg:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/opp/op_impl/built-in/ai_core/tbe && export LD_LIBRARY_PATH=/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/lib64:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/driver/lib64:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/add-ons:/usr/local/python3.7.5/lib:/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/acllib/lib64 && export SLOG_PRINT_TO_STDOUT=1 && export ASCEND_OPP_PATH=/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/opp && /home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/bin/atc --input_shape="s:1,42,42,1;W_0/actor/c_in:1,256;W_0/actor/h_in:1,256" --check_report=/home/HwHiAiUser/modelzoo/alien_model/device/network_analysis.report --input_format=NHWC --output="/home/HwHiAiUser/modelzoo/alien_model/device/alien_model" --soc_version=Ascend310 --framework=3 --model="/home/HwHiAiUser/AscendProjects/A3C/model/alien_model.pb" --enable_scope_fusion_passes=ScopeDynamicRNNPass

ATC start working now, please wait for a moment.

Traceback (most recent call last):

File "/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/atc/python/site-packages/te/platform/fusion_manager.py", line 706, in call_op_func

return opfunc(*inputs, *outputs, *attrs)

File "/home/HwHiAiUser/Ascend/ascend-toolkit/20.1.rc1.alpha001/opp/op_impl/built-in/ai_core/tbe/impl/mat_mul.py", line 227, in check_supported

check_shape(shape_a, param_name="input_x1")

File "/home/HwHiAiUser/Ascend/ascend-toolkit/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值