From 3d1ff70da4e93aa0f5ff9a7e8cbd153b487a1868 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 25 Feb 2023 16:05:21 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc2/.decode.py.swp | Bin 65536 -> 65536 bytes .../ASR/conformer_ctc2/.transformer.py.swp | Bin 57344 -> 57344 bytes egs/librispeech/ASR/conformer_ctc2/decode.py | 8 ++++++++ 3 files changed, 8 insertions(+) diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index 7f6616eff9e54e89f1be04b9d6a518112ac94cf9..b47ef3dee4055f148be465532786a9d4571c5332 100644 GIT binary patch delta 396 zcmZo@U}dByofC8;U#a3LKAn7cr9Zfb6RQDuB?VsSRAL~34g zeoAUlJVXIPF0rH}J}1ApSix2yHy5ZPM=vE6qB1SN25a%>C?3<3-c-kSvr*crXr*cljFf!Gs>m4R3Z zh_3>b%mU&FAhrTxOCSbm+sVqnupWrJfw&5YRe@Lrt}UBj|DsW6{;GxJ|b#>w~o kNb@i#KtXnDF(aF#L2yZqkox(VIU8U&eN9A%I0F_oG3jhEB diff --git a/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp b/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp index 9311babfdd7da6fdb4893f5ae6a97e6c4d3c3c86..6634a7b61787e2a4a6691c7fe5f98266570a51fd 100644 GIT binary patch delta 32 mcmZoTz}#?vSv1KY%+puFQqO<^2m}}ycsBe@PTVN^`aS@Zp$X~$ delta 32 mcmZoTz}#?vSv1KY%+puFQqO<^2m}}yKCb_nthZ70^?d-FrwW?@ diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py index b3d0e6038..8b38b0908 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -415,6 +415,14 @@ def decode_one_batch( return {key: hyps} if params.method == "greedy-search": + att_loss = mmodel.decoder_forward( + encoder_memory, + memory_mask, + token_ids=unsorted_token_ids, + sos_id=graph_compiler.sos_id, + eos_id=graph_compiler.eos_id, + warmup=warmup, + ) hyps = greedy_search(nnet_output, memory_key_padding_mask) # hyps is a list of str, e.g., ['xxx yyy zzz', ...]