diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py
index e74d8faa6f7c48d1ea91f47cbe47d2d3cf5bf704..004322c832195fdbe5cbdaf20f4186ba7e9f8a26 100644
--- a/ppocr/utils/save_load.py
+++ b/ppocr/utils/save_load.py
@@ -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]
             else:
                 param_state_dict[key] = model_dict[key]
-        model.set_dict(param_state_dict)
+        model.set_state_dict(param_state_dict)
         return
 
-    param_state_dict, optim_state_dict = paddle.load(path)
-    model.set_dict(param_state_dict)
+    param_state_dict = paddle.load(path + '.pdparams')
+    model.set_state_dict(param_state_dict)
     return
 
 
@@ -91,7 +91,7 @@ def init_model(config, model, logger, optimizer=None, lr_scheduler=None):
             "Given dir {}.pdopt not exist.".format(checkpoints)
         para_dict = paddle.load(checkpoints + '.pdparams')
         opti_dict = paddle.load(checkpoints + '.pdopt')
-        model.set_dict(para_dict)
+        model.set_state_dict(para_dict)
         if optimizer is not None:
             optimizer.set_state_dict(opti_dict)