Update save_load.py
This commit is contained in:
parent
e0f3bcb494
commit
a4abf1322c
|
@ -115,8 +115,8 @@ def init_model(config, program, exe):
|
||||||
logger.info("Finish initing model from {}".format(path))
|
logger.info("Finish initing model from {}".format(path))
|
||||||
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:
|
||||||
|
|
Loading…
Reference in New Issue