From b30add8ae51e51cc085750fa7f3a5f27a7f6f5ce Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Thu, 28 Jan 2021 11:05:28 +0800
Subject: [PATCH] fix bug

---
 tools/export_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/export_model.py b/tools/export_model.py
index a2428bf7..542dffe3 100755
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -47,7 +47,7 @@ def main():
         char_num = len(getattr(post_process_class, 'character'))
         config['Architecture']["Head"]['out_channels'] = char_num
     model = build_model(config['Architecture'])
-    # init_model(config, model, logger)
+    init_model(config, model, logger)
     model.eval()
 
     save_path = '{}/inference'.format(config['Global']['save_inference_dir'])
-- 
GitLab