From cbc1ad4f10f6fc425b566c58f42d4a634860e1c6 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 2 Feb 2023 18:59:31 +0900
Subject: [PATCH] from local

---
 .../ASR/transformer_ctc/.decode.py.swp        | Bin 36864 -> 40960 bytes
 egs/aishell/ASR/transformer_ctc/decode.py     |   6 ++++++
 2 files changed, 6 insertions(+)

diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
index 276f8a4b65da8869d09a5ae5eb6d3e99b76ab002..e82de0b0c276b689443fcdae85935db315ea7e21 100644
GIT binary patch
delta 474
zcmYMw&r8B!7zglY)1XjuO4%WT7m3=ShC!EtNTEYfL3D_~B;3?k%9L$EAd#mEf*`yl
zMP9mf(A@%|L;pn-bgDy8m!M13chl;D5AToX#|v-elCRwI5hb>^u*|097$F3r*3rE>
zeAS4&2D$F4N4VycYC{+Z_c&=pe|sT`w{L`zV3T=S;G8E)c#%N6Vu#{oZlj6g1suZ`
zIO#ad!l#Gm0nQ)~tI!2rc*Ceis6z%+7>5wNaP7D4uf#gPPplJGF`v~pGn#583!0uy
zPNY(7hPm5V>aWT<cdljAGPIH?$!4jbS)%0D<cfyd!!(`crroRm!DuY+wac(Ama@87
zRp$GK+zL~+cCF|biyBc$HnQoA^OPw{_WPW$xSh@r)%--hK>L4w|L;3U^b9pP1r?TH
k5C~emL?^HVB8);9IJm>CLpXpHm;_(IeiG__{LKgd1ED5kJpcdz

delta 182
zcmZoTz|^pSNi4}A%+puFQqO<^2m}}ymUY}tezkO?*c*OE_Q}iwYZ(hCZxj&U{6av2
zf8qg!O$sauys|)*e;Fal3pWcoR5J$MWMN?F1>$BP4h3R6Am#<)Ps|Jq>w&ldh|_^s
z3Wz0vm<5PGGcho11L8^`t^i^uAXWonkkMZmCvFtuQ~-g*%;JpA+;hJ&ZT{rK=M4Zi
C^(Yzu

diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py
index 46873fe11..cd9069766 100755
--- a/egs/aishell/ASR/transformer_ctc/decode.py
+++ b/egs/aishell/ASR/transformer_ctc/decode.py
@@ -548,6 +548,12 @@ def main():
         num_decoder_layers=params.num_decoder_layers,
     )
 
+    f = open(f"{params.lang_dir}/tokens.txt", "r").readlines()
+    token_dict = {}
+    for line in f:
+        line = line.strip().split()
+        token_dict[line[1]] = line[0]
+
     if params.avg == 1:
         load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model)
     else: