Skip to content

P
PaddleOCR

PaddlePaddle / PaddleOCR
1 年多 前同步成功

通知 1532
Star 32963
Fork 6643
P
PaddleOCR
提交 4d775dc9 编写于 作者: W WenmuZhou
浏览文件
操作

rc版本适配

上级 44840726
隐藏空白更改
内联 并排
Showing with 4 addition and 4 deletion
ppocr/utils/save_load.py
浏览文件 @ 4d775dc9
...@@ -68,11 +68,11 @@ def load_dygraph_pretrain(model, logger, path=None, load_static_weights=False): ...@@ -68,11 +68,11 @@ def load_dygraph_pretrain(model, logger, path=None, load_static_weights=False):
param_state_dict[key] = pre_state_dict[weight_name] param_state_dict[key] = pre_state_dict[weight_name]
else: else:
param_state_dict[key] = model_dict[key] param_state_dict[key] = model_dict[key]
model.set_dict(param_state_dict) model.set_state_dict(param_state_dict)
return return
param_state_dict, optim_state_dict = paddle.load(path) param_state_dict = paddle.load(path + '.pdparams')
model.set_dict(param_state_dict) model.set_state_dict(param_state_dict)
return return
...@@ -91,7 +91,7 @@ def init_model(config, model, logger, optimizer=None, lr_scheduler=None): ...@@ -91,7 +91,7 @@ def init_model(config, model, logger, optimizer=None, lr_scheduler=None):
"Given dir {}.pdopt not exist.".format(checkpoints) "Given dir {}.pdopt not exist.".format(checkpoints)
para_dict = paddle.load(checkpoints + '.pdparams') para_dict = paddle.load(checkpoints + '.pdparams')
opti_dict = paddle.load(checkpoints + '.pdopt') opti_dict = paddle.load(checkpoints + '.pdopt')
model.set_dict(para_dict) model.set_state_dict(para_dict)
if optimizer is not None: if optimizer is not None:
optimizer.set_state_dict(opti_dict) optimizer.set_state_dict(opti_dict)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或