Skip to content

Commit 69ffcc4

Browse files
author
chengduozh
committed
fix io doc
test=develop
1 parent cc9028b commit 69ffcc4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/io.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ def name_has_fc(var):
145145
146146
prog = fluid.default_main_program()
147147
fluid.io.save_vars(executor=exe, dirname=path, main_program=prog,
148-
vars=None)
148+
vars=None, predicate = name_has_fc)
149149
# All variables in `main_program` whose name includes "fc" will be saved.
150150
# And variables are going to be saved separately.
151151
@@ -369,7 +369,7 @@ def name_has_fc(var):
369369
370370
prog = fluid.default_main_program()
371371
fluid.io.load_vars(executor=exe, dirname=path, main_program=prog,
372-
vars=None)
372+
vars=None, predicate=name_has_fc)
373373
# All variables in `main_program` whose name includes "fc" will be loaded.
374374
# And all the variables are supposed to have been saved in differnet files.
375375

0 commit comments

Comments
 (0)