fix trainer load parameters (#12816)

dataset_flowers_-_change_md5
chengduo 7 years ago committed by GitHub
parent 8f9f414a14
commit 3ae97aab6c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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,

Loading…
Cancel
Save