Skip to content

GitLab

"...composable_kernel_onnx.git" did not exist on "cac014f17355d6504b618f5945c6326a285db7e9"
Unverified Commit 3a2d0f61 authored by bingooo's avatar bingooo Committed by GitHub
Browse files

fix bug

parent 5661c686
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/utils/save_load.py
View file @ 3a2d0f61
...@@ -98,7 +98,7 @@ def load_dygraph_params(config, model, logger, optimizer): ...@@ -98,7 +98,7 @@ def load_dygraph_params(config, model, logger, optimizer):
pm = config['Global']['pretrained_model'] pm = config['Global']['pretrained_model']
if pm is None: if pm is None:
return {} return {}
if not os.path.exists(pm) or not os.path.exists(pm + ".pdparams"): if not os.path.exists(pm) and not os.path.exists(pm + ".pdparams"):
logger.info(f"The pretrained_model {pm} does not exists!") logger.info(f"The pretrained_model {pm} does not exists!")
return {} return {}
pm = pm if pm.endswith('.pdparams') else pm + '.pdparams' pm = pm if pm.endswith('.pdparams') else pm + '.pdparams'
......
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