fix error in padding computing (#1607)

k2-fsa · Apr 25, 2024 · 25cabb7 · 25cabb7
1 parent df36f93
commit 25cabb7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/librispeech/SSL/zipformer/hubert_ce.py b/egs/librispeech/SSL/zipformer/hubert_ce.py
@@ -429,7 +429,7 @@ def forward(
         # padding_mask: (B, T), bool
         # mask_indices: (B, T), bool
         x = x.transpose(0, 1)
-        x, x_lens = self.encoder(x, ~padding_mask.sum(dim=-1))
+        x, x_lens = self.encoder(x, (~padding_mask).sum(dim=-1))
         x = x.transpose(0, 1)
 
         if features_only: