Fix bug in conv asr

AI4Bharat · Apr 22, 2024 · 7a625e1 · 7a625e1
1 parent c48c9b8
commit 7a625e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/asr/modules/conv_asr.py b/nemo/collections/asr/modules/conv_asr.py
@@ -478,8 +478,8 @@ def forward(self, encoder_output, language_ids=None): #CTEMO
             mask = mask.to(decoder_output.device)
             # masked_output = self.masked_softmax(decoder_output, mask) # B x T x 3073 -> B x T x 257
             decoder_output = torch.masked_select(decoder_output, mask).view(decoder_output.shape[0],decoder_output.shape[1],-1)
-        
-        del mask
+
+            del mask
         # print(mask[0][0])
         # softmax_output = self.masked_softmax(decoder_output, mask)
         # return softmax_output