From 2d981114b3a736a07407abdf384e5bf927d14692 Mon Sep 17 00:00:00 2001 From: yuyang18 Date: Fri, 20 Jul 2018 19:04:22 +0800 Subject: [PATCH] Skip reader when save --- python/paddle/fluid/io.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index cf43998228..e1d26474e6 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -66,7 +66,8 @@ def is_persistable(var): res = fluid.io.is_persistable(param) """ if var.desc.type() == core.VarDesc.VarType.FEED_MINIBATCH or \ - var.desc.type() == core.VarDesc.VarType.FETCH_LIST: + var.desc.type() == core.VarDesc.VarType.FETCH_LIST or \ + var.desc.type() == core.VarDesc.VarType.READER: return False return var.persistable