diff --git a/egs/libriheavy/LM/zipformer1/subformer.py b/egs/libriheavy/LM/zipformer1/subformer.py index d4b761a0a..baf0ed914 100644 --- a/egs/libriheavy/LM/zipformer1/subformer.py +++ b/egs/libriheavy/LM/zipformer1/subformer.py @@ -820,7 +820,7 @@ class LearnedDownsamplingModule(nn.Module): self.score_balancer = Balancer(1, channel_dim=-1, min_positive=1/(2*downsampling_factor), max_positive=0.6, - min_abs=1.0) + min_abs=1.0) # below are for diagnostics. self.copy_weights1 = nn.Identity()