add max_text_length to export model

2021-04-28 13:41:04 +08:00 · 2021-04-28 13:41:04 +08:00 · d36c19ce1a
parent b566bcbbaa
commit d36c19ce1a
1 changed files with 1 additions and 1 deletions
--- 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'])