diff --git a/tools/export_model.py b/tools/export_model.py index 37e7ba61314e8e74cc27f33b3e8349bbc8531b9a..bdff89f755d465742f1c2a810f8ae76153a558c6 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'])