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

Increase max_factor in final lm_balancer and am_balancer

Browse Source
This commit is contained in:
Daniel Povey 2022-10-22 20:02:54 +08:00
parent 8b3bba9b54
commit 7a55cac346
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/librispeech/ASR/pruned_transducer_stateless7/model.py
View File

@ -72,11 +72,13 @@ class Transducer(nn.Module):
self.lm_balancer = ActivationBalancer(vocab_size,
channel_dim=-1,
min_positive=0, max_positive=1,
min_abs=0, max_abs=50.0)
min_abs=0, max_abs=50.0,
max_factor=0.1)
self.am_balancer = ActivationBalancer(vocab_size,
channel_dim=-1,
min_positive=0, max_positive=1,
min_abs=0, max_abs=20.0)
min_abs=0, max_abs=30.0,
max_factor=0.1)
def forward(
self,