From 082d326d659796bf696290264e4e74a93a1ce1a1 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 25 Feb 2023 19:38:41 +0900
Subject: [PATCH] from local

---
 icefall/.decode.py.swp | Bin 65536 -> 65536 bytes
 icefall/decode.py      |   8 +++-----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/icefall/.decode.py.swp b/icefall/.decode.py.swp
index 3a026c86dcd6894798a1f013c06f8ccb9dd26527..cf4815b7135c53fa02e461767cc9d41a7294d22f 100644
GIT binary patch
delta 331
zcmXBPF-yZ>6oug%NZLkQG@zB}Qqe^;S`iBNA1HJP7QsOjvpETZ4i3$rlY_%INI?)>
zGCAm{lTZqRqa=e!GKxqS!AWpXe8mF~TsX_QVQ7Y-S>CFyE;TBPvsEF3A_q@j!Q|dm
zx$YRA_8r|R-4#TR?f=$OP8_*MeIiXbc%=DjTiDe}d{Iz-I;7UpF}0al>MVUTCX}DK
zI?C3Et<#)Hr0kECR<k5Bj?bb<59ip1hZ(%~i`?S|8rNv!5IbmK9wT^RM}jUcae@Ff
z6wzZ_j1UVLwPj9RjsXY~UhHBY0cx1SdtT%gQC=HnJ7-pi?j#$QGQX$w^0RH|`p<*$
E3(9Or%m4rY

delta 352
zcmXZXJxBs^7{~GFLQl^+NJx8Gk{W`fPRk{iW{Cz7f*>$!aca^fsE!7=rauuuErB%D
zkp1_A2pUs|Uc4y^Tv{4h8kD|J4}6~If#32xt{ZUOKzu8;mRn6M&8LJK7Fl@u+5V_D
z;@PO-=zLUr_E}h@V*RC#incDtYY~xM<Y3~S)I~+NTNh!YsPdgD)mk=H`PESy{)oE%
zpL71ETJKDdS-S2QW5Hunq^_*5!a>0nnZ);yNEhcQBZ*nO4~pF377g6s1c%tc3LM1n
z#!XFlxWW-i$RL6)w?1MYX^dIo$S8}AKJPZ+p^6eRn89aAM57+kM&Q&ONF=n^E(Vp^
TOB&NE-CLyZy*%q||H1eL+&@!c

diff --git a/icefall/decode.py b/icefall/decode.py
index 53ab39ecc..000570787 100644
--- a/icefall/decode.py
+++ b/icefall/decode.py
@@ -956,13 +956,11 @@ def rescore_with_attention_decoder(
     attention_scores = -nll.sum(dim=1)
     
     if ngram_lm_scale is None:
-        ngram_lm_scale_list = [0.]
-        '''
+        #ngram_lm_scale_list = [0.]
         ngram_lm_scale_list = [0.01, 0.05, 0.08]
         ngram_lm_scale_list += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0]
         ngram_lm_scale_list += [1.1, 1.2, 1.3, 1.5, 1.7, 1.9, 2.0]
         ngram_lm_scale_list += [2.1, 2.2, 2.3, 2.5, 3.0, 4.0, 5.0]
-        '''
     else:
         ngram_lm_scale_list = [ngram_lm_scale]
 
@@ -980,8 +978,8 @@ def rescore_with_attention_decoder(
     for n_scale in ngram_lm_scale_list:
         for a_scale in attention_scale_list:
             tot_scores = (
-                #am_scores.values
-                #+ n_scale * ngram_lm_scores.values
+                am_scores.values
+                + n_scale * ngram_lm_scores.values
                 a_scale * attention_scores
             )
             ragged_tot_scores = k2.RaggedTensor(nbest.shape, tot_scores)