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

Bug fix

Browse Source
This commit is contained in:
Daniel Povey 2023-03-31 17:23:25 +08:00
parent cd0f48f508
commit c2e39bd488
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
View File

@ -301,7 +301,7 @@ class Zipformer2(EncoderInterface):
max_downsampling_factor = max(downsampling_factor) max_downsampling_factor = max(downsampling_factor)
num_frames_max = (num_frames0 + max_downsampling_factor - 1) num_frames_max = (num_frames0 + max_downsampling_factor - 1) // max_downsampling_factor
# we divide the dropped-out feature dimensions into two equal groups; # we divide the dropped-out feature dimensions into two equal groups;
# the first group is dropped out with probability 0.1, the second # the first group is dropped out with probability 0.1, the second
@ -325,7 +325,7 @@ class Zipformer2(EncoderInterface):
feature_masks = [] feature_masks = []
for i in range(num_encoders): for i in range(num_encoders):
ds = downsampling_factor[i] ds = self.downsampling_factor[i]
upsample_factor = (max_downsampling_factor // ds) upsample_factor = (max_downsampling_factor // ds)
frame_mask = (frame_mask_max.unsqueeze(1).expand(num_frames_max, upsample_factor, frame_mask = (frame_mask_max.unsqueeze(1).expand(num_frames_max, upsample_factor,