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

Use downsampling_factor ** -0.333 as the scale for stacks

Browse Source
This commit is contained in:
Daniel Povey 2023-01-05 14:23:40 +08:00
parent 0d7161ebec
commit 95e8296014
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

@ -209,7 +209,7 @@ class Zipformer(EncoderInterface):
) )
# we are adding a new attribute here. # we are adding a new attribute here.
# this will be interpreted by get_named_parameter_groups_with_lrs(). # this will be interpreted by get_named_parameter_groups_with_lrs().
encoder.lr_scale = downsampling_factor[i] ** -0.5 encoder.lr_scale = downsampling_factor[i] ** -0.333
encoders.append(encoder) encoders.append(encoder)
self.encoders = nn.ModuleList(encoders) self.encoders = nn.ModuleList(encoders)