mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 06:34:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
jinzr 2023-11-30 23:06:20 +08:00
parent 4f697c4caa
commit 26d387c41f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/vctk/TTS/vits/train.py
View File

@ -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()