From 81a73e32f58176eb25cae3fb9e1d123a69b6c031 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 26 Apr 2023 12:18:32 +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 0ccd9b774aff273c49a447dbb32369118e0fb260..769eac71a93f3d5823a14a0719c13e2c97c59280 100644 GIT binary patch delta 52 zcmZoTz|?SnNi@kI%+puFQqO<^2m}}y{!RBt`Myz<+k;O>Aum6#Dzzv-+8|a(VY8mc GrEma}5fB>y delta 52 zcmZoTz|?SnNi@kI%+puFQqO<^2m}}yE=>1GIlNJn+k-DQIxj!3Dzzv-+8{PIdb6I# GrEma}Nf40$ 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 97a21c4a0..17596e55f 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 @@ -495,7 +495,7 @@ class Data2VecAudioModel(BaseFairseqModel): nonzero = padding_mask[i].nonzero() try: print(nonzero[0]) - conv_feat_all = torch.cat([conv_feat_all, features[i][nonzero[0]][:]]) + 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())