mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 15:44:17 +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:03 +08:00
parent 48c90df9c9
commit 4f697c4caa
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -684,6 +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,
) )
audio_pred = audio_pred.data.cpu().numpy() audio_pred = audio_pred.data.cpu().numpy()
audio_len_pred = ( audio_len_pred = (