Update utility.py

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):