diff --git a/tools/export_model.py b/tools/export_model.py index a2428bf720c04c160e443c989683a018f33e20eb..542dffe360cb5f65dbcc82617a31b4b73a6f06c7 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'])