improve save_persistable api doc. test=develop (PaddlePaddle#15911)

colourful-tree · Mar 1, 2019 · 46c5e37 · 46c5e37
1 parent 31d830d
commit 46c5e37
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py
@@ -468,9 +468,10 @@ def save_persistables(executor, dirname, main_program=None, filename=None):
 
             exe = fluid.Executor(fluid.CPUPlace())
             param_path = "./my_paddle_model"
+            # `prog` can be a program defined by the user
             prog = fluid.default_main_program()
             fluid.io.save_persistables(executor=exe, dirname=param_path,
-                                       main_program=None)
+                                       main_program=prog)
     """
 
     if main_program and main_program._is_distributed: