diff options
-rw-r--r-- | stanza/models/constituency_parser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/stanza/models/constituency_parser.py b/stanza/models/constituency_parser.py index 1c588e01..f6d71823 100644 --- a/stanza/models/constituency_parser.py +++ b/stanza/models/constituency_parser.py @@ -509,7 +509,7 @@ def main(args=None): elif args['mode'] == 'predict': trainer.evaluate(args, model_load_file, retag_pipeline) elif args['mode'] == 'remove_optimizer': - trainer.remove_optimizer(args, model_save_file, model_load_file) + trainer.remove_optimizer(args, args['save_name'], model_load_file) if __name__ == '__main__': main() |