From 0ba3c62e350322b4ab2eefd7f44c620fad0edaa1 Mon Sep 17 00:00:00 2001
From: Jason <jiangjiajun@baidu.com>
Date: Fri, 1 Feb 2019 11:33:50 +0800
Subject: [PATCH] Update convert.py

---
 tensorflow2fluid/src/convert.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tensorflow2fluid/src/convert.py b/tensorflow2fluid/src/convert.py
index 5c1c374..2a1bfe6 100644
--- a/tensorflow2fluid/src/convert.py
+++ b/tensorflow2fluid/src/convert.py
@@ -38,6 +38,9 @@ def _convert(args):
         for i in range(len(items)):
             if items[i] != "None":
                 items[i] = int(items[i])
+            else:
+                items[i] = None
+                
         input_shape.append(items)
 
     sys.stderr.write("\nLoading tensorflow model......\n")
-- 
GitLab