From d36c19ce1a1b7431df1b65ab95ee0c7f2296bcde Mon Sep 17 00:00:00 2001 From: WenmuZhou 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'])