From d36c19ce1a1b7431df1b65ab95ee0c7f2296bcde Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Wed, 28 Apr 2021 13:41:04 +0800
Subject: [PATCH] add max_text_length to export model

---
 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 37e7ba61..bdff89f7 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