Merge pull request #12286 from reyoung/feature/skip_reader_when_save

Skip reader when save
7 years ago · 1e417b93e0
parent 02cf54d331 2d981114b3
commit 1e417b93e0
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