Unverified Commit a4abf132 authored by shaohua.zhang's avatar shaohua.zhang Committed by GitHub
Browse files

Update save_load.py

parent e0f3bcb4
...@@ -116,7 +116,7 @@ def init_model(config, program, exe): ...@@ -116,7 +116,7 @@ def init_model(config, program, exe):
else: else:
raise ValueError( raise ValueError(
"Model checkpoints {} does not exists," "Model checkpoints {} does not exists,"
"check if you lost the file prefix .".format(checkpoints + '.pdparams')) "check if you lost the file prefix.".format(checkpoints + '.pdparams'))
pretrain_weights = config['Global'].get('pretrain_weights') pretrain_weights = config['Global'].get('pretrain_weights')
if pretrain_weights: if pretrain_weights:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment