From 2d71992d80ba4142d2d5a5df17c69c2f2ac553fd Mon Sep 17 00:00:00 2001
From: 34j <55338215+34j@users.noreply.github.com>
Date: Sun, 16 Apr 2023 17:57:03 +0900
Subject: [PATCH] fix(train): fix y_mel length (#356)

---
 src/so_vits_svc_fork/train.py | 1 +
 1 file changed, 1 insertion(+)

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,