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 RE seq lengths

Browse Source
This commit is contained in:
Daniel Povey 2022-10-28 19:26:06 +08:00
parent d7d5188bd9
commit de9a6ebd6c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -228,7 +228,8 @@ class Zipformer(EncoderInterface):
x = self.downsample_output(x)
# class Downsample has this rounding behavior..
lengths = (x_lens + 1) // 2
assert self.output_downsampling_factor == 2
lengths = (lengths + 1) // 2
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)