提交 7973d9b4 编写于 作者: T tangwei12

bug fix

上级 46f2688f
......@@ -178,10 +178,11 @@ class Trainer(object):
exe.run(self.startup_program)
if self.checkpoint and self.checkpoint.load_serial:
exe = executor.Executor(place)
io.load_checkpoint(exe, self.checkpoint.checkpoint_dir,
self.checkpoint.load_serial,
self.startup_program)
with self._prog_and_scope_guard():
exe = executor.Executor(place)
io.load_checkpoint(exe, self.checkpoint.checkpoint_dir,
self.checkpoint.load_serial,
self.startup_program)
if not self.checkpoint.is_pserver:
epoch_id, step_id = io.load_trainer_args(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册