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:
+                try:
                    shape = graph_node.out_shapes[0]
                    if len(shape) > 0 and shape.count(-1) < 2:
                        need_define_shape = 0
+                except:
+                    pass

            if need_define_shape > 0:
                shape = None