Skip to content

X
X2Paddle

PaddlePaddle / X2Paddle
大约 1 年 前同步成功

通知 328
Star 698
Fork 167
X
X2Paddle
提交 6c3cb861 编写于 作者: J jiangjiajun
浏览文件
操作

fix bug

上级 8ec74138
显示空白变更内容
内联 并排
Showing with 6 addition and 3 deletion
x2paddle/decoder/tf_decoder.py
浏览文件 @ 6c3cb861
...@@ -341,9 +341,12 @@ class TFDecoder(object): ...@@ -341,9 +341,12 @@ class TFDecoder(object):
need_define_shape = 2 need_define_shape = 2
if need_define_shape == 1: if need_define_shape == 1:
try:
shape = graph_node.out_shapes[0] shape = graph_node.out_shapes[0]
if len(shape) > 0 and shape.count(-1) < 2: if len(shape) > 0 and shape.count(-1) < 2:
need_define_shape = 0 need_define_shape = 0
except:
pass
if need_define_shape > 0: if need_define_shape > 0:
shape = None shape = None
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或