feat: least amount of changes

EveryVoiceTTS · Sep 13, 2024 · 04d9d0d · 04d9d0d
1 parent 1092bb7
commit 04d9d0d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/everyvoice/base_cli/helpers.py b/everyvoice/base_cli/helpers.py
@@ -242,7 +242,7 @@ def train_base_command(
     # This callback will always save the last checkpoint
     # regardless of its performance.
     last_ckpt_callback = ModelCheckpoint(
-        save_top_k=-1,
+        save_top_k=1,
         save_last=True,
         every_n_train_steps=config.training.ckpt_steps,
         every_n_epochs=config.training.ckpt_epochs,
@@ -277,8 +277,8 @@ def train_base_command(
         # NOTE: If we don't check validation at the end of the training epoch, saving the last checkpoint is disabled.
         # pytorch_lightning/callbacks/model_checkpoint.py :  431
         # _should_save_on_train_epoch_end [ModelCheckpoint]
-        # check_val_every_n_epoch=config.training.check_val_every_n_epoch,
-        check_val_every_n_epoch=1,
+        check_val_every_n_epoch=config.training.check_val_every_n_epoch,
+        # check_val_every_n_epoch=1,
         val_check_interval=config.training.val_check_interval,
         callbacks=[
             monitored_ckpt_callback,