From 6d686fe729a0b75bbc91d1bc1665a511b7dbdcd8 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Wed, 26 Apr 2023 12:15:12 +0900
Subject: [PATCH] from local

---
 .../.data2vec_audio.py.swp                    | Bin 40960 -> 40960 bytes
 .../data2vec_audio.py                         |   6 +++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp
index 95186d28d91cfe337ab85d2e2f342b359d7a1914..78ad58f06e0a955ac1b8b1352e26c9e836d571f0 100644
GIT binary patch
delta 272
zcmZoTz|?SnNi@kI%+puFQqO<^2m}}y7EJR<G2JLyEx_oud7?nEKCc}s149uTMAB`t
zpu=}H1`Y;>73>TQ6M)zqh~<I!El|Z_Af5!otw3DFwpq{7hjp^Q`^HJ0Y`n?&d1diw
zsfi`=i8(ow``tw+d9Y00pv*G4(?iTGIk7||8byVU0!VFXQEG9lCW8VRNUcatEhw>?
o{IEq7q}>Uh_UO#m=)C;As??(VXoJ{TET)wdRa$Kp@puys0Gka=ZvX%Q

delta 152
zcmZoTz|?SnNi@kI%+puFQqO<^2m}}yW=``+;n^r!Ex_osd7?nEKCdzx1A{v|MAB)p
zpu=~yFF@WQAf5rl=|CI>#2^(;Kx_=ePgofkuCQ*_bM#@I%n>U(d3Ee;Sp^VCt;j3^
YGoS>J4`xj+a@#q1hr7sT5sx?F0P-#)djJ3c

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
index ee504fb21..e63a8a2ee 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
@@ -496,7 +496,11 @@ class Data2VecAudioModel(BaseFairseqModel):
             conv_feat_all = torch.tensor([])
             for i in range(padding_mask.size()[0]):
                 nonzero = padding_mask[i].nonzero()
-                conv_feat_all = 
+                try:
+                    conv_feat_all = torch.cat([conv_feat_all, features[i][nonzero[0]]])
+                except:
+                    conv_feat_all = torch.cat([conv_feat_all, features])
+            print(conv_feat_all.size())
                 #print(padding_mask[i].nonzero())
             print(padding_mask.nonzero())
             print(padding_mask.nonzero().size())