From 673f41d2bf3dd8c891c775a1d16dad7f4534c830 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 26 Apr 2023 12:18:22 +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 aeb01696fdd939ab0eaa4318d657ab2f14a355f4..0ccd9b774aff273c49a447dbb32369118e0fb260 100644 GIT binary patch delta 81 zcmZoTz|?SnNi@kI%+puFQqO<^2m}}yE=>1GIlNKyy*;Zu8v}#&WKM_Qf;ZS17!Ck& gIS~5+vGrzAM$-sop61-kw#7je)^#GN;3D!Q1Q%42OZZ d3Wx)N*lx3^qY>NWKW<`^queiVHt@I*4gfF`73}~3 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 7b076d0a3..97a21c4a0 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())