fix bug

2021-07-01 15:41:02 +08:00 · 2021-07-01 15:41:02 +08:00 · 61cee7e663
parent 3a2d0f61ee
commit 61cee7e663
1 changed files with 1 additions and 1 deletions
--- a/ppocr/utils/save_load.py
+++ b/ppocr/utils/save_load.py
@ -91,7 +91,7 @@ def init_model(config, model, optimizer=None, lr_scheduler=None):

 def load_dygraph_params(config, model, logger, optimizer):
    ckp = config['Global']['checkpoints']
-    if ckp and os.path.exists(ckp):
+    if ckp and os.path.exists(ckp + ".pdparams"):
        pre_best_model_dict = init_model(config, model, optimizer)
        return pre_best_model_dict
    else: