diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py
index c6d20651283954000241f80a28d22c7821af2ff0..e74d8faa6f7c48d1ea91f47cbe47d2d3cf5bf704 100644
--- a/ppocr/utils/save_load.py
+++ b/ppocr/utils/save_load.py
@@ -89,7 +89,8 @@ def init_model(config, model, logger, optimizer=None, lr_scheduler=None):
             "Given dir {}.pdparams not exist.".format(checkpoints)
         assert os.path.exists(checkpoints + ".pdopt"), \
             "Given dir {}.pdopt not exist.".format(checkpoints)
-        para_dict, opti_dict = paddle.load(checkpoints)
+        para_dict = paddle.load(checkpoints + '.pdparams')
+        opti_dict = paddle.load(checkpoints + '.pdopt')
         model.set_dict(para_dict)
         if optimizer is not None:
             optimizer.set_state_dict(opti_dict)
@@ -133,8 +134,8 @@ def save_model(net,
     """
     _mkdir_if_not_exist(model_path, logger)
     model_prefix = os.path.join(model_path, prefix)
-    paddle.save(net.state_dict(), model_prefix)
-    paddle.save(optimizer.state_dict(), model_prefix)
+    paddle.save(net.state_dict(), model_prefix + '.pdparams')
+    paddle.save(optimizer.state_dict(), model_prefix + '.pdopt')
 
     # save metric and config
     with open(model_prefix + '.states', 'wb') as f: