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

Fix bug

Browse Source
This commit is contained in:
Daniel Povey 2023-03-29 20:33:43 +08:00
parent f1dbf4222e
commit a02199df79
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/ASR/pruned_transducer_stateless7/zipformer.py
View File

@ -332,7 +332,7 @@ class Zipformer2(EncoderInterface):
.reshape(num_frames_max * upsample_factor, batch_size, 3)) .reshape(num_frames_max * upsample_factor, batch_size, 3))
num_frames = (num_frames0 + ds - 1) // ds num_frames = (num_frames0 + ds - 1) // ds
frame_mask = frame_mask[:num_frames] frame_mask = frame_mask[:num_frames]
channels = self.encoder_unmasked_dim[i] channels = self.encoder_dim[i]
feature_mask = torch.ones(num_frames, batch_size, channels, feature_mask = torch.ones(num_frames, batch_size, channels,
dtype=x.dtype, device=x.device) dtype=x.dtype, device=x.device)
u1 = self.encoder_unmasked_dim[i] u1 = self.encoder_unmasked_dim[i]