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

---
 .../.data2vec_audio.py.swp                    | Bin 40960 -> 40960 bytes
 .../data2vec_audio.py                         |   2 +-
 2 files changed, 1 insertion(+), 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 78ad58f06e0a955ac1b8b1352e26c9e836d571f0..3633323aee861318af7fc0334613bdccf7d81d60 100644
GIT binary patch
delta 119
zcmZoTz|?SnNi@kI%+puFQqO<^2m}}y9Hx7uT-qr5-kw#1m4P8<GN;35!*mV?24f)p
z&(6T`3Wz5GaTO370x>5LUuR=rSPR58K%5W6F`IQA^;tI;xtTLf4vH6^{J}$P@_i4U
J%_lr8!U1uh9OVE2

delta 119
zcmZoTz|?SnNi@kI%+puFQqO<^2m}}y7EJR<G2JNo-k#Ntm4TsXGN;35!&(jo22UUs
z1!4vuUct`5Fae0&fmj}h-?A|<90uY^K->z%MVoaU^;swHj}@BS70a_Z#Z8}S@)|e!
J%_lr8!T}{e8_ECx

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 e63a8a2ee..116fcc2fd 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
@@ -493,7 +493,7 @@ class Data2VecAudioModel(BaseFairseqModel):
             print(features.size())
             print(padding_mask.size())
             print(padding_mask[0])
-            conv_feat_all = torch.tensor([])
+            conv_feat_all = torch.tensor([]).to(features.device())
             for i in range(padding_mask.size()[0]):
                 nonzero = padding_mask[i].nonzero()
                 try: