From 26d387c41f62000027d6d4e0407110a1b42d953a Mon Sep 17 00:00:00 2001 From: jinzr Date: Thu, 30 Nov 2023 23:06:20 +0800 Subject: [PATCH] Update train.py --- egs/vctk/TTS/vits/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/vctk/TTS/vits/train.py b/egs/vctk/TTS/vits/train.py index 538373d60..59ecbf194 100755 --- a/egs/vctk/TTS/vits/train.py +++ b/egs/vctk/TTS/vits/train.py @@ -683,7 +683,7 @@ def compute_validation_loss( if batch_idx == 0 and rank == 0: inner_model = model.module if isinstance(model, DDP) else model audio_pred, _, duration = inner_model.inference( - text=tokens[0, : tokens_lens[0].item()] + text=tokens[0, : tokens_lens[0].item()], sids=speakers, ) audio_pred = audio_pred.data.cpu().numpy()