mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
jinzr 2024-03-01 09:40:43 +08:00
parent a0dc097ad9
commit 8f8e516e29
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

@ -342,7 +342,7 @@ def prepare_input(
torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).int().to(device) torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).int().to(device)
) )
tokens = tokenizer.texts_to_token_ids( tokens = tokenizer.tokens_to_token_ids(
tokens, intersperse_blank=True, add_sos=True, add_eos=True tokens, intersperse_blank=True, add_sos=True, add_eos=True
) )
tokens = k2.RaggedTensor(tokens) tokens = k2.RaggedTensor(tokens)