From 74c5476f68822e696e0a65ff86f7080306febffc Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Wed, 25 Jul 2018 15:12:59 +0800
Subject: [PATCH] hidden slice_vars in distribute transpile, hidden it to users

---
 python/paddle/fluid/io.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py
index e5eb34eb0f5..f2d9666afa9 100644
--- 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):
     """
-- 
GitLab