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

fix trainer load parameters (#12817)

上级 fa1feb45
...@@ -285,6 +285,7 @@ class Trainer(object): ...@@ -285,6 +285,7 @@ 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):
with self._prog_and_scope_guard():
# load params from param_path into scope # load params from param_path into scope
io.load_persistables( io.load_persistables(
executor=exe, executor=exe,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册