diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py index 1d760e983a635dcc6b48b839ee99434c67b4378d..df599706a67adf7d8c42d7917b87039295f7e934 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 - os.path.exists(pretrained + '.pdparams')): + not os.path.exists(pretrained + '.pdparams')): raise ValueError("Model pretrain path {} does not " "exists.".format(pretrained)) param_state_dict = paddle.load(pretrained + '.pdparams')