Skip to content

GitLab

Unverified Commit f7111d7b authored by littletomatodonkey's avatar littletomatodonkey Committed by GitHub
Browse files

Merge pull request #467 from BeyondYourself/develop 

add a necessary check for checkpoints
parents f7b5edcc a4abf132
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 3 deletions
ppocr/utils/save_load.py
View file @ f7111d7b
...@@ -109,9 +109,14 @@ def init_model(config, program, exe): ...@@ -109,9 +109,14 @@ def init_model(config, program, exe):
""" """
checkpoints = config['Global'].get('checkpoints') checkpoints = config['Global'].get('checkpoints')
if checkpoints: if checkpoints:
path = checkpoints if os.path.exists(checkpoints + '.pdparams'):
fluid.load(program, path, exe) path = checkpoints
logger.info("Finish initing model from {}".format(path)) fluid.load(program, path, exe)
logger.info("Finish initing model from {}".format(path))
else:
raise ValueError(
"Model checkpoints {} does not exists,"
"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