From 754d259a9418b4d2f6bf55d68d1041d582b2fe81 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Wed, 15 Feb 2023 14:45:04 +0900
Subject: [PATCH] from local

---
 egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 69632 -> 69632 bytes
 egs/tedlium2/ASR/conformer_ctc3/train.py      |  21 +++++++-----------
 2 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
index 0d721c82fad48e2634e555d5a252ae05adac2e4e..b7b7a1b9b80f810bb68077c13c348f336346cf67 100644
GIT binary patch
delta 351
zcmW;H!7Bt|7{~Ev)}-0BYiHU$iHwpJ#ilvP38z)7t^EfkRyGlb-IK&--6ZlBQ8Z_p
zqT%4;;H)esIjldxg@fA^-`S@=y-!a)zn-_|=`~NkX_{J6y_)@+<^Pze&z-4+$T`Fm
zv3SBdON#8n#VmT!>K19@1r2PWfC2on|9}Rr5au0V6JdS~?{Se^Y@+}NMqK>*_g$CD
zmZK^iSf{O+7FDTWkU^Y5Mj64l@^U);5PcTY=<BfUjBf<Xk>bLV$RzPc2*1Fk?fRp6
z#32^pz`!%-o}df|o%mq)5*O{9`=U0Nh}fd?yHC2ivR+j)Z)w$>A65DKm%2XAsoGwT
UWslga6Jw=n$z9v1RE9<V0n;T<;{X5v

delta 341
zcmX}mzb^xE7{~GFn>4}wa$(WjP>W!Y_y=^CSlrUdP(*SPCDqN)OT^NF@D-<%6N81O
z%`q6vrV5FP!OVb;7WF=nKFKT3Gd$n#^9kcj7-yOft}^tan^RuWafe^4p()aDiHIxW
z(b9Sv6uChS1x(||5xK`MwopPJzHN~<ny6t8dCVY-cmBS`9+ohUEJncHWc%yNj2gCT
z>C1yfeR4OX<D+%G&^=4ej|<8fr8dezsWP#$S)A0J`jlT>7Fp$T18HjGFSAIOlrOkI
z74sOw8+*@a|Gzn<A3|dXW#`&gvv!m>RVB(<Y9RVF)LQgmDaT9C_bTX33_;S`_th`J
CKuDbc

diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py
index 1152610e6..870eec1bc 100755
--- a/egs/tedlium2/ASR/conformer_ctc3/train.py
+++ b/egs/tedlium2/ASR/conformer_ctc3/train.py
@@ -650,24 +650,19 @@ def compute_loss(
                 allow_truncate=params.subsampling_factor - 1,
             )
 
-            ctc_loss = k2.ctc_loss(
+            ctc_loss = (1-params.interctc_weight) * k2.ctc_loss(
                 decoding_graph=decoding_graph,
                 dense_fsa_vec=dense_fsa_vec,
                 output_beam=params.beam_size,
                 reduction=params.reduction,
                 use_double_scores=params.use_double_scores,
-            ) 
-            
-            inter_ctc_loss = 0
-            for fsa in dense_fsa_vec_inter:
-                print(fsa)
-                inter_ctc_loss += k2.ctc_loss(
-                    decoding_graph=decoding_graph,
-                    dense_fsa_vec=fsa,
-                    output_beam=params.beam_size,
-                    reduction=params.reduction,
-                    use_double_scores=params.use_double_scores,
-                ) 
+            ) + params.interctc_weight * k2.ctc_loss(
+                decoding_graph=decoding_graph,
+                dense_fsa_vec=dense_fsa_vec_inter,
+                output_beam=params.beam_size,
+                reduction=params.reduction,
+                use_double_scores=params.use_double_scores,
+
             
             ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss
         else: