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

Fix convert_texts_into_ids() in the tedlium3 recipe. (#1929)

Browse Source
This commit is contained in:
Fangjun Kuang 2025-04-24 17:04:46 +08:00 committed by GitHub
parent 5ec95e5482
commit cc2e64a6aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/tedlium3/ASR/pruned_transducer_stateless/train.py
View File

@ -422,7 +422,7 @@ def compute_loss(
texts = batch["supervisions"]["text"]
unk_id = params.unk_id
y = convert_texts_into_ids(texts, unk_id, sp=sp)
y = convert_texts_into_ids(texts, sp=sp)
y = k2.RaggedTensor(y).to(device)
with torch.set_grad_enabled(is_training):

2
egs/tedlium3/ASR/transducer_stateless/train.py
View File

@ -397,7 +397,7 @@ def compute_loss(
texts = batch["supervisions"]["text"]
unk_id = params.unk_id
y = convert_texts_into_ids(texts, unk_id, sp=sp)
y = convert_texts_into_ids(texts, sp=sp)
y = k2.RaggedTensor(y).to(device)
with torch.set_grad_enabled(is_training):