Update train.py

2025-09-04 06:34:20 +00:00 · 2023-11-30 23:06:03 +08:00 · 2023-11-30 23:06:03 +08:00 · 4f697c4caa
commit 4f697c4caa
parent 48c90df9c9
1 changed files with 1 additions and 0 deletions
--- a/egs/vctk/TTS/vits/train.py
+++ b/egs/vctk/TTS/vits/train.py
@ -684,6 +684,7 @@ def compute_validation_loss(
                inner_model = model.module if isinstance(model, DDP) else model
                audio_pred, _, duration = inner_model.inference(
                    text=tokens[0, : tokens_lens[0].item()]
+                    sids=speakers,
                )
                audio_pred = audio_pred.data.cpu().numpy()
                audio_len_pred = (