Increase default grad_scale of Balancer from 0.02 to 0.04.

This commit is contained in:
Daniel Povey 2022-12-29 21:38:19 +08:00
parent 59be36181c
commit e164393e91

View File

@ -1078,7 +1078,7 @@ class Balancer(torch.nn.Module):
max_positive: FloatLike = 0.95,
min_abs: FloatLike = 0.2,
max_abs: FloatLike = 100.0,
grad_scale: FloatLike = 0.02,
grad_scale: FloatLike = 0.04,
prob: Optional[FloatLike] = None,
):
super().__init__()