mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-25 16:10:17 +09:00
parent e44db1dd76
commit e173f765e8
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
View File

Binary file not shown.

2
egs/librispeech/ASR/conformer_ctc2/decode.py
View File

@ -419,7 +419,7 @@ def decode_one_batch(
unsorted_token_ids = graph_compiler.texts_to_ids(supervisions["text"]) unsorted_token_ids = graph_compiler.texts_to_ids(supervisions["text"])
att_loss = model.decoder_forward( att_loss = model.decoder_forward(
memory, memory,
momory_key_padding_mask, memory_key_padding_mask,
token_ids=unsorted_token_ids, token_ids=unsorted_token_ids,
sos_id=graph_compiler.sos_id, sos_id=graph_compiler.sos_id,
eos_id=graph_compiler.eos_id, eos_id=graph_compiler.eos_id,