未验证 提交 eabe89de 编写于 作者: Y yukavio 提交者: GitHub

fix prune_io (#629)

上级 87a3ffd8
......@@ -31,7 +31,7 @@ def save_model(exe, graph, dirname):
_logger.info("Save model weights into {}".format(weights_file))
shapes = {}
for var in graph.program.list_vars():
if var.persistable:
if var.persistable and str(var.type) != 'VarType.READER':
shapes[var.name] = var.shape
SHAPES_FILE = os.path.join(dirname, _SHAPES_FILE)
if not os.path.exists(dirname):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册