fix error in load model

2026-09-24 23:33:08 +08:00 · 2021-07-15 16:08:56 +08:00
parent 697719ba3c
commit f7a62d945e
1 changed files with 1 additions and 1 deletions
@@ -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')