hidden slice_vars in distribute transpile, hidden it to users

7 years ago · 74c5476f68
parent 19ddd3da6a
commit 74c5476f68
1 changed files with 4 additions and 5 deletions
--- a/python/paddle/fluid/io.py
+++ b/python/paddle/fluid/io.py
@ -401,13 +401,12 @@ def load_vars(executor,
                inputs={},
                outputs={"Out": load_var_list},
                attrs={'file_path': os.path.join(dirname, filename)})
-
-        if main_program._slice_vars_and_atts:
-            _load_slice_up_vars(executor, dirname,
-                                main_program._slice_vars_and_atts)
-
        executor.run(load_prog)

+        # load slice vars on pserver, if have it.
+        _load_slice_up_vars(executor, dirname,
+                            main_program._slice_vars_and_atts)
+

 def load_params(executor, dirname, main_program=None, filename=None):
    """