fix(train): don't save model when tuning for auto batch size

voicepaw · Apr 19, 2023 · 2311a35 · 2311a35
1 parent 1ddb7ed
commit 2311a35
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py
@@ -254,7 +254,8 @@ def stft(
             torch.stft = stft
 
     def on_train_end(self) -> None:
-        self.save_checkpoints(adjust=0)
+        if not self.tuning:
+            self.save_checkpoints(adjust=0)
 
     def save_checkpoints(self, adjust=1):
         # `on_train_end` will be the actual epoch, not a -1, so we have to call it with `adjust = 0`
@@ -546,5 +547,5 @@ def validation_step(self, batch, batch_idx):
             )
 
     def on_validation_end(self) -> None:
-        if not self.trainer.sanity_checking:
+        if not self.trainer.sanity_checking and not self.tuning:
             self.save_checkpoints()