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 01d980981..3f3b39f45 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp differ
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 73653ff63..c0fb4caca 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,10 @@ class Data2VecAudioModel(BaseFairseqModel):
             padding_mask = None
         
         print(padding_mask.size())
-        print(len((padding_mask == True).nonzero(as_tuple=True)[0]))
+        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])
         exit()
 
         if self.post_extract_proj is not None: