From c4ca35a251e974f52011191842fca3f4c5ad7a29 Mon Sep 17 00:00:00 2001
From: "shaohua.zhang" <BeyondYourself@users.noreply.github.com>
Date: Mon, 27 Jul 2020 17:11:33 +0800
Subject: [PATCH] Update utility.py

---
 ppocr/utils/utility.py | 3 +++
 1 file changed, 3 insertions(+)

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