Skip to content

GitLab

Commit d36c19ce authored by WenmuZhou's avatar WenmuZhou
Browse files

add max_text_length to export model

parent b566bcbb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/export_model.py
View file @ d36c19ce
...@@ -47,7 +47,7 @@ def main(): ...@@ -47,7 +47,7 @@ def main():
char_num = len(getattr(post_process_class, 'character')) char_num = len(getattr(post_process_class, 'character'))
config['Architecture']["Head"]['out_channels'] = char_num config['Architecture']["Head"]['out_channels'] = char_num
model = build_model(config['Architecture']) model = build_model(config['Architecture'])
# init_model(config, model, logger) init_model(config, model, logger)
model.eval() model.eval()
save_path = '{}/inference'.format(config['Global']['save_inference_dir']) save_path = '{}/inference'.format(config['Global']['save_inference_dir'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment