opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-10-29 18:48:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

change is to does

Browse Source
This commit is contained in:
WenmuZhou
2021-11-29 08:38:15 +00:00
parent 389e315ae0
commit 7c15d1b790
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ppocr/utils/save_load.py
+2 -2
View File
@@ -57,7 +57,7 @@ def load_model(config, model, optimizer=None):
if checkpoints.endswith('.pdparams'):
checkpoints = checkpoints.replace('.pdparams', '')
assert os.path.exists(checkpoints + ".pdparams"), \
"The {}.pdparams is not exists!".format(checkpoints)
"The {}.pdparams does not exists!".format(checkpoints)
# load params from trained model
params = paddle.load(checkpoints + '.pdparams')
@@ -106,7 +106,7 @@ def load_pretrained_params(model, path):
if path.endswith('.pdparams'):
path = path.replace('.pdparams', '')
assert os.path.exists(path + ".pdparams"), \
"The {}.pdparams is not exists!".format(path)
"The {}.pdparams does not exists!".format(path)
params = paddle.load(path + '.pdparams')
state_dict = model.state_dict()