diff --git a/ppocr/utils/utility.py b/ppocr/utils/utility.py
index 74a200f5e5661ecfe1409290871a931bdf18e99d..4d23f62656f5561dd93f40fa97a3f7874e4b2040 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):