Skip reader when save

7 years ago · 2d981114b3
parent 5ebbfaa8b9
commit 2d981114b3
1 changed files with 2 additions and 1 deletions
--- 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