未验证 提交 3ae97aab 编写于 作者: C chengduo 提交者: GitHub

fix trainer load parameters (#12816)

上级 8f9f414a
...@@ -285,11 +285,12 @@ class Trainer(object): ...@@ -285,11 +285,12 @@ class Trainer(object):
self._load_checkpoint() self._load_checkpoint()
if param_path and os.path.isdir(param_path): if param_path and os.path.isdir(param_path):
# load params from param_path into scope with self._prog_and_scope_guard():
io.load_persistables( # load params from param_path into scope
executor=exe, io.load_persistables(
dirname=param_path, executor=exe,
main_program=self.startup_program) dirname=param_path,
main_program=self.startup_program)
def _transpile_nccl2_dist(self): def _transpile_nccl2_dist(self):
# PADDLE_TRAINER_IPS # PADDLE_TRAINER_IPS
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册