mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-09 05:55:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge b4c9e37c8e7e15ef14546b3f38debb8b3dfa6933 into 0904e490c5fb424dc5cb4d14ae468e4d32a07dc4

Browse Source
This commit is contained in:
jianyou 2025-11-28 16:35:13 -05:00 committed by GitHub
commit 1b17e38928
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/SSL/zipformer/hubert_ce.py
View File

@ -429,7 +429,7 @@ class HubertModel(nn.Module):
# 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) if padding_mask is not None else None)
x = x.transpose(0, 1)
if features_only: