未验证 提交 0ba3c62e 编写于 作者: J Jason 提交者: GitHub

Update convert.py

上级 82dcc01b
...@@ -38,6 +38,9 @@ def _convert(args): ...@@ -38,6 +38,9 @@ def _convert(args):
for i in range(len(items)): for i in range(len(items)):
if items[i] != "None": if items[i] != "None":
items[i] = int(items[i]) items[i] = int(items[i])
else:
items[i] = None
input_shape.append(items) input_shape.append(items)
sys.stderr.write("\nLoading tensorflow model......\n") sys.stderr.write("\nLoading tensorflow model......\n")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册