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:09:11 +08:00
parent 26d387c41f
commit 8c75259723
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

@ -684,7 +684,7 @@ def compute_validation_loss(
inner_model = model.module if isinstance(model, DDP) else model inner_model = model.module if isinstance(model, DDP) else model
audio_pred, _, duration = inner_model.inference( audio_pred, _, duration = inner_model.inference(
text=tokens[0, : tokens_lens[0].item()], text=tokens[0, : tokens_lens[0].item()],
sids=speakers, sids=speakers[0],
) )
audio_pred = audio_pred.data.cpu().numpy() audio_pred = audio_pred.data.cpu().numpy()
audio_len_pred = ( audio_len_pred = (