From be5f456790dd1c4718dd023609276a334e0278cc Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Fri, 9 Jun 2023 16:59:08 +0900
Subject: [PATCH] from local

---
 .../.train_tta.py.swp                         | Bin 77824 -> 81920 bytes
 .../train_tta.py                              |   3 +--
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp
index 155f33296794828e41bf13b53f79196bc5c2332a..e4b82d3c6412d3986faede05f6cf641046c4dfe8 100644
GIT binary patch
delta 250
zcmZp8z|zpbx`9c6QD`!=z*)xh$rlC0g@hR(fD1~yGH<>mz{bBxfu(`>8Y2TkA2U!w
zfPo=>v*3c+jGEhlT#!%&5XS+r9uVJPVqoY6;%Fd_0%AKLW}JL=q4wkl6FE0?EJ|dY
z{KHlSh&5RhfME0P#hi7FHq#F(GL}!i(8w}9T8UA7@`dHOo4q#u=b3(3pV8JSF@=Fa
zi;saJ4~Rjw!(8`_pMl{F5YGW(Hz58Els^f?CxCbj5a$CiNKY=`W<{a<yb~YzZCW&u
F5dhQ~J0}1D

delta 157
zcmZo@U~PE7vVlo}QE)P|z*)wk$rlC0H**O7;G6hBVUq$&1Ftj#1H*qtAR)lOP_$Wa
z!E8p&-OLOOlYn?45SIgSED-Ag@pUE!h8`e}0^&#@ww-)+q4wkpjVzm47o{>z=3T@+
y{joixEF;6@zQvlGcQ59wn{1M#GVwt^NX>KyB}Vbdua^gH_S*EHXZm4%Mq2<6@-Hg@

diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
index 2d4d1b0e4..0f6e67176 100755
--- a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
+++ b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
@@ -832,9 +832,8 @@ def compute_loss(
     # at entry, feature is (N, T, C)
     assert feature.ndim == 2 or feature.ndim == 3
     feature = feature.to(device)
-
+    
     supervisions = batch["supervisions"]
-
     if feature.ndim == 2:
         feature_lens = []
         for supervision in supervisions['cut']: