Reduce min_prob of ActivationBalancer from 0.1 to 0.05.

This commit is contained in:
Daniel Povey 2022-10-21 15:42:04 +08:00
parent c5cb52fed1
commit 476fb9e9f3

View File

@ -482,7 +482,7 @@ class ActivationBalancer(torch.nn.Module):
scale_gain_factor: float = 0.02,
min_abs: float = 0.2,
max_abs: float = 100.0,
min_prob: float = 0.1,
min_prob: float = 0.05,
):
super(ActivationBalancer, self).__init__()
self.num_channels = num_channels