From f7a62d945eebf85ecbeb9228ac74f6742e89bdaa Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Thu, 15 Jul 2021 16:08:56 +0800 Subject: [PATCH] fix error in load model --- ppocr/utils/save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py index df599706..1d760e98 100644 --- a/ppocr/utils/save_load.py +++ b/ppocr/utils/save_load.py @@ -77,7 +77,7 @@ def init_model(config, model, optimizer=None, lr_scheduler=None): pretrained_model = [pretrained_model] for pretrained in pretrained_model: if not (os.path.isdir(pretrained) or - not os.path.exists(pretrained + '.pdparams')): + os.path.exists(pretrained + '.pdparams')): raise ValueError("Model pretrain path {} does not " "exists.".format(pretrained)) param_state_dict = paddle.load(pretrained + '.pdparams') -- GitLab