fix bug

61cee7e6 · bingooo · GitHub · 3a2d0f61 · 61cee7e6
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppocr/utils/save_load.py ppocr/utils/save_load.py +1 -1

未找到文件。
--- 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: