Skip to content

Commit a8afb98

Browse files
Merge pull request #16928 from wanghaoshuang/cp_slim_ck
[cherry-pick] Fix load persistables in graph wrapper.
2 parents 3063449 + c8e49b6 commit a8afb98

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/paddle/fluid/contrib/slim/graph/graph_wrapper.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -489,8 +489,11 @@ def load_persistables(self, path, exe):
489489
def if_exist(var):
490490
return os.path.exists(os.path.join(path, var.name))
491491

492-
io.load_vars(
493-
exe.exe, path, vars=self.persistables.values(), predicate=if_exist)
492+
persistables = []
493+
for var in self.persistables.values():
494+
if if_exist(var):
495+
persistables.append(var)
496+
io.load_vars(exe.exe, path, vars=persistables, predicate=if_exist)
494497

495498
def update_param_shape(self, scope):
496499
"""

0 commit comments

Comments
 (0)