From a31218ada2de4f9329c769f15b08c6e179512fa0 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 25 Feb 2023 16:22:53 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc2/.decode.py.swp | Bin 73728 -> 73728 bytes egs/librispeech/ASR/conformer_ctc2/decode.py | 3 +-- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index efa0ffee49f2f1b4004a6c11f2b846eebb4b674a..228f23539faf008a13b7a6e29a5ddf2943042fc2 100644 GIT binary patch delta 235 zcmZoTz|wGlMJ&l6%+puFQqO<^2m}}yZf^UT938k(%u0aKcypk@W`AB5E(V4-Kn#*K zo-CLUUVn=hs1Asa1MylQZUo|3Aa(#^9U#^QVl5yR2I6l#3=H>x_yiEI1mgKXoC?I2 zK&%7AqCm_G#5_QJlbeBI77)(_;&vd;1!6}amIY!4AO^YOH5bF?=LsBKo8M)LF-t~U p=~(3^R>WuKrKDEq6=znZY8Yz9>L?gY4$Kpsd?#Cfb5h>CjR1iZG4ucc delta 223 zcmZoTz|wGlMJ&l6%+puFQqO<^2m}}y61M(Kt_;{HW+lLAvN=#-vp?^DP6me8To754 z$$|;t^%8sx3^#ch7;XUZF(6(8#0@|k1H|?~tPR9kK&%PGLO}eLhk@ZP5FZEP6+k=> zh*N;r0*JMNSOkc9fS4PIZ*Vg(%mm^YK->nzIY8_H#4@a44b9EEd6zc=0Hi4|MgRZ+ diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py index 478f60625..fb5b0295b 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -238,8 +238,7 @@ def greedy_search( """ ctc_probs = probs _, max_index = ctc_probs.max(2) # (B, maxlen) - print(mask.size()) - max_index = max_index.masked_fill_(mask, 0) # (B, maxlen) + max_index = max_index.masked_fill_(mask[:,:max_index.size(1)], 0) # (B, maxlen) ret_hyps = [] for hyp in max_index: