fix(train): fix y_mel length (#356)

voicepaw · Apr 16, 2023 · 2d71992 · 2d71992
1 parent f7f3c9b
commit 2d71992
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py
@@ -370,6 +370,7 @@ def training_step(self, batch: dict[str, torch.Tensor], batch_idx: int) -> None:
             self.hparams.train.segment_size // self.hparams.data.hop_length,
         )
         y_hat_mel = mel_spectrogram_torch(y_hat.squeeze(1), self.hparams)
+        y_mel = y_mel[..., : y_hat_mel.shape[-1]]
         y = commons.slice_segments(
             y,
             ids_slice * self.hparams.data.hop_length,