diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py
index 4e1b02e4..546487ad 100644
--- 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,