From 1bd60745ee905eb24c4bce6f2c27d13347cdbac1 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 14 Feb 2023 18:00:42 +0900
Subject: [PATCH] from local

---
 egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 61440 -> 61440 bytes
 egs/tedlium2/ASR/conformer_ctc3/train.py      |  35 ++++++++++++++----
 2 files changed, 28 insertions(+), 7 deletions(-)

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
index 318e8b357d5af4851458575b905cc0491dcc8e6f..09fef4cf4bd26b9870d5f619361e6dcd2441eea6 100644
GIT binary patch
delta 621
zcmbW!Pbh<N7zgm@^}g9Zn^&`ye{C0k7VRJN@1RI&hqWn5X%xG;IJn!vi7f3ok}`9W
zqH*9rDJQa&axvZu4z#shI1t}$QOaR?>eKJ-*YiAYzvl@nVptK)Q_fMZ&)Q>m5<(^#
z@jXws=l0A;vY_QF*&;X+S=o3p|EeNNs*5>}d@ka+l4u*2AOyYeW*~ZiGgyZ?aDxRt
z^+Y$Y12ZrN-5|qb1yLNLumFQ#f+ro(p^p5tjE`ZcgBLA+4l59X7N~_9(80Zi=oHpq
z0s`QNR(QuFlInyYeW-ZWEpcQ>%@=V?!gjkTa`>)72y&pSnI|)SEO$zLnQkha%2s8y
zt0UM(*5AQGtf4%AWozTC*YtOt>pyiC5nW4YZ}gPX-5yn>-XNB+0r^E^Ooh0{()J+&
z0T=)em?4MU^FmFcUzs-i%<FPh!^w2x_h3YKu=x10SUrEF^lZ&Y!Z=sO4tRG{;^vg`
E3yr{+umAu6

delta 283
zcmZp8z})bFSuDvQ%+puFQqO<^2m}}y^nG6^fBv{pY?lBd=j4k5@{H1(KMDx%uP@_f
zVE6~bAT`n;wLox)pMhZu5N`(JVjy+_Vlg0o!N<U`2Z$Sh*b9iIf%q*i1H*nGo(#m{
zK<o*`9zbjb#C$;fk%xidG!XX!aXk>%0dW)%YXdRR-px!b)!dT{j94c3xpPcDvqXOL
zuNz&=ll^b{PI|y%RtL0T9}t5a#|d^C1H)yYLqHDe1!8|7b_8M*AZ7>RCt&rP9Umm{
TO+26iH+Jt`t<4XX@-PAb9t=J-

diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py
index 3b5a3b728..9ba5ca6b2 100755
--- a/egs/tedlium2/ASR/conformer_ctc3/train.py
+++ b/egs/tedlium2/ASR/conformer_ctc3/train.py
@@ -583,18 +583,39 @@ def compute_loss(
             )
 
         elif params.condition:
-            dense_fsa_vec1 = k2.DenseFsaVec(
+            dense_fsa_vec = k2.DenseFsaVec(
                 nnet_output[0],
                 supervision_segments,
                 allow_truncate=params.subsampling_factor - 1,
             )
             
-            dense_fsa_vec2 = k2.DenseFsaVec(
-                nnet_output[1][8],
-                supervision_segments,
-                allow_truncate=params.subsampling_factor - 1,
-            )
-
+            dense_fsa_vec2 = [
+                    k2.DenseFsaVec(
+                    nnet_output[1][2],
+                    supervision_segments,
+                    allow_truncate=params.subsampling_factor - 1,
+                ),
+                k2.DenseFsaVec(
+                    nnet_output[1][5],
+                    supervision_segments,
+                    allow_truncate=params.subsampling_factor - 1,
+                ),
+                k2.DenseFsaVec(
+                    nnet_output[1][8],
+                    supervision_segments,
+                    allow_truncate=params.subsampling_factor - 1,
+                ),
+                k2.DenseFsaVec(
+                    nnet_output[1][11],
+                    supervision_segments,
+                    allow_truncate=params.subsampling_factor - 1,
+                ),
+                k2.DenseFsaVec(
+                    nnet_output[1][14],
+                    supervision_segments,
+                    allow_truncate=params.subsampling_factor - 1,
+                )
+            ]
             ctc_loss = (1-params.interctc_weight)* k2.ctc_loss(
                 decoding_graph=decoding_graph,
                 dense_fsa_vec=dense_fsa_vec1,