From 889c919048d67cef9668d859b1935d77f9bca731 Mon Sep 17 00:00:00 2001 From: Jeff Wang Date: Mon, 7 May 2018 11:50:16 -0700 Subject: [PATCH] Use _prog_and_scope_guard to switch the scope (#10421) --- python/paddle/fluid/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/trainer.py b/python/paddle/fluid/trainer.py index 1cbecd69e5..d44cb16bfb 100644 --- a/python/paddle/fluid/trainer.py +++ b/python/paddle/fluid/trainer.py @@ -173,9 +173,9 @@ class Trainer(object): def save_params(self, param_path): # reference: save_persistables in io.py - exe = executor.Executor(self.place) - io.save_persistables( - exe, dirname=param_path, main_program=self.startup_program) + with self._prog_and_scope_guard(): + exe = executor.Executor(self.place) + io.save_persistables(exe, dirname=param_path) @staticmethod def _check_and_get_place(place): -- GitLab