We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 5d04a9b + 5747123 commit d4ea377Copy full SHA for d4ea377
flair/trainers/trainer.py
@@ -845,7 +845,7 @@ def wrapped_forward_loss(*args, **kwargs2):
845
846
# if we do not use dev data for model selection, save final model
847
if save_final_model:
848
- self._save_model(base_path / "final-model.pt", save_optimizer_state == save_optimizer_state)
+ self._save_model(base_path / "final-model.pt", save_optimizer_state=save_optimizer_state)
849
850
except KeyboardInterrupt:
851
log_line(log)
0 commit comments