From a6f20c4b49b21ecfce5d1d3a2335894c503f9769 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 11 Apr 2023 15:32:58 +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 288955dc0594d7e01703967e63272e726cc355d8..304274668b89662a34df533828f550f9ab54099a 100644
GIT binary patch
delta 108
zcmZoTz|?SnNi4}A%+puFQqO<^2m}}y#Q&S5bWYkR_RgM_i<N=FXfmh6Yuy$O1_pm1
ze!<SbFb|050&xKl=L4}f5VHgE0X7DPN+34ctmr7oI{BoV=;i>o2~3;sdnkqj08NM(
A%>V!Z

delta 108
zcmZoTz|?SnNi4}A%+puFQqO<^2m}}yRR5Ty@bqmIduPwe&C0-FJekwswQeg114951
zzhq}%m=DDBfVdEd3xL=Mh&h1xAR7Zi6%ZS5R&<nPoqXI)Y_q@H1g6aoJQTtKD%Tie

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 d2513b298..25d1c8699 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
@@ -491,7 +491,7 @@ class Data2VecAudioModel(BaseFairseqModel):
         ## for prompt tuning
         if prompt is not None:
             #features = torch.cat([features, prompt])
-            prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1])
+            prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))
             features = torch.cat([prompt, features])
             prompt_padding_mask = torch.zeros(promt.size()).type(torch.BoolTensor).to(features.device)
             padding_mask = torch.cat([prompt_padding_mask, padding_mask])