fix bug

6c3cb861 · jiangjiajun · 8ec74138 · 6c3cb861
隐藏空白更改
内联并排

Showing with 6 addition and 3 deletion

x2paddle/decoder/tf_decoder.py x2paddle/decoder/tf_decoder.py +6 -3

未找到文件。
--- a/x2paddle/decoder/tf_decoder.py
+++ b/x2paddle/decoder/tf_decoder.py
@@ -341,9 +341,12 @@ class TFDecoder(object):
                    need_define_shape = 2
            if need_define_shape == 1:
-                shape = graph_node.out_shapes[0]
+                try:
-                if len(shape) > 0 and shape.count(-1) < 2:
+                    shape = graph_node.out_shapes[0]
-                    need_define_shape = 0
+                    if len(shape) > 0 and shape.count(-1) < 2:
+                        need_define_shape = 0
+                except:
+                    pass
            if need_define_shape > 0:
                shape = None