From d01264cb9b19e9994f292f58761bbce1e1f3c2df Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 14:58:55 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 40960 -> 40960 bytes .../data2vec_audio.py | 8 ++++---- 2 files changed, 4 insertions(+), 4 deletions(-) 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 3f3b39f45c0880c2c7cf6122aa925393955d5a58..1bc205f83aae0c35f571b2fdd7ef78d26d32c3fd 100644 GIT binary patch delta 99 zcmZoTz|?SnNi4}A%+puFQqO<^2m}}yK7TVwk!;;4_RgL)hmC=uWiqG3Z(d&x1_mx5 tKF$sl*(~Uo&o)_7i+A#Icj3uX+*KztdB{z^1*V@*M6&3zt4;Q(969M%8; delta 95 zcmZoTz|?SnNi4}A%+puFQqO<^2m}}y_Ixu*dD^m3?43Pp0viKE#bi#0-@INN3=Awl oypJ6yvRTkEpKWrehsfjw?y8eLJQOEChSF*tAimq?ZjaJ%02X;1kpKVy 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 c0fb4caca..13c20a989 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,10 +495,10 @@ class Data2VecAudioModel(BaseFairseqModel): padding_mask = None print(padding_mask.size()) - print((padding_mask[0] == True).nonzero(as_tuple=True)[0][0]) - print((padding_mask[1] == True).nonzero(as_tuple=True)[0][0]) - print((padding_mask[2] == True).nonzero(as_tuple=True)[0][0]) - print((padding_mask[3] == True).nonzero(as_tuple=True)[0][0]) + print((padding_mask[0] == True).nonzero(as_tuple=True)[0]) + print((padding_mask[1] == True).nonzero(as_tuple=True)[0]) + print((padding_mask[2] == True).nonzero(as_tuple=True)[0]) + print((padding_mask[3] == True).nonzero(as_tuple=True)[0]) exit() if self.post_extract_proj is not None: