diff options
| author | John Bauer <horatio@gmail.com> | 2022-09-12 09:26:56 +0300 |
|---|---|---|
| committer | John Bauer <horatio@gmail.com> | 2022-09-12 09:26:56 +0300 |
| commit | 8dc42113931992245112caf510c99c5cade2599c (patch) | |
| tree | 88936c0ac1844afc1f33a4f469fa56ec8a59ed3f | |
| parent | c4d785729e42ac90f298e0ef4ab487d14fa35591 (diff) | |
Fix remove_optimizer mode
| -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() |