diff --git a/ppocr/utils/utility.py b/ppocr/utils/utility.py
index 74a200f5..4d23f626 100755
--- a/ppocr/utils/utility.py
+++ b/ppocr/utils/utility.py
@@ -21,6 +21,7 @@ import os
 import shutil
 import tempfile
 
+import paddle
 import paddle.fluid as fluid
 
 from .utility import initial_logger
@@ -112,12 +113,14 @@ def init_model(config, program, exe):
         path = checkpoints
         fluid.load(program, path, exe)
         logger.info("Finish initing model from {}".format(path))
+        return
 
     pretrain_weights = config['Global'].get('pretrain_weights')
     if pretrain_weights:
         path = pretrain_weights
         load_params(exe, program, path)
         logger.info("Finish initing model from {}".format(path))
+    return
 
 
 def save_model(program, model_path):