Skip to content
Snippets Groups Projects

Merge develop to master

Merged Mateusz Klimaszewski requested to merge develop into master
Viewing commit c0835180
Show latest version
3 files
+ 6
5
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -16,6 +16,7 @@ class FinishingTrainingCheckpointer(training.Checkpointer):
epoch: Union[int, str],
trainer: "allen_trainer.Trainer",
is_best_so_far: bool = False,
save_model_only: bool = False,
) -> None:
if trainer._learning_rate_scheduler.decreases <= 1 or epoch == trainer._num_epochs - 1:
super().save_checkpoint(epoch, trainer, is_best_so_far)