From 9d80f13024ba189991b03834acd4b126ae9ebcfc Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 26 Apr 2023 12:18:45 +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 769eac71a93f3d5823a14a0719c13e2c97c59280..618f58ed8b019d36e316895f2a3f434b19bbfc47 100644 GIT binary patch delta 81 zcmZoTz|?SnNi4}A%+puFQqO<^2m}}y*k^d8D4*FV_RgMFj*WrAYBHz8Z^7&A3=I2$ gxD1GWf!J!ZsG}j<f=yZ`_I delta 79 zcmZoTz|?SnNi4}A%+puFQqO<^2m}}y{!RBt`F?t%*gJbxc{T&cuBzXflwGcX(g e;&LGN17hpVqK<}almEGiOpbIvv02IEUN``Qs~1oJ 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 17596e55f..3c0a7f2f7 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())