fix trainer load parameters (#12816)

7 years ago · 3ae97aab6c
parent 8f9f414a14
commit 3ae97aab6c
1 changed files with 6 additions and 5 deletions
--- a/python/paddle/fluid/trainer.py
+++ b/python/paddle/fluid/trainer.py
@ -285,11 +285,12 @@ class Trainer(object):
            self._load_checkpoint()

        if param_path and os.path.isdir(param_path):
-            # load params from param_path into scope
-            io.load_persistables(
-                executor=exe,
-                dirname=param_path,
-                main_program=self.startup_program)
+            with self._prog_and_scope_guard():
+                # load params from param_path into scope
+                io.load_persistables(
+                    executor=exe,
+                    dirname=param_path,
+                    main_program=self.startup_program)

    def _transpile_nccl2_dist(self):
        # PADDLE_TRAINER_IPS