Skip to content

Commit d4ea377

Browse files
authored
Merge pull request #3674 from flairNLP/save_final_model
Fix minor bug in saving final model
2 parents 5d04a9b + 5747123 commit d4ea377

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

flair/trainers/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -845,7 +845,7 @@ def wrapped_forward_loss(*args, **kwargs2):
845845

846846
# if we do not use dev data for model selection, save final model
847847
if save_final_model:
848-
self._save_model(base_path / "final-model.pt", save_optimizer_state == save_optimizer_state)
848+
self._save_model(base_path / "final-model.pt", save_optimizer_state=save_optimizer_state)
849849

850850
except KeyboardInterrupt:
851851
log_line(log)

0 commit comments

Comments
 (0)