Increase grad_scale in score_balancer

This commit is contained in:
Daniel Povey 2023-05-29 10:56:01 +08:00
parent 79f1863a1e
commit d950496d5a

View File

@ -875,7 +875,8 @@ class LearnedDownsamplingModule(nn.Module):
min_positive=1/(2*downsampling_factor),
max_positive=0.6,
min_abs=1.0,
max_abs=4.0)
max_abs=4.0,
grad_scale=0.1)
# below are for diagnostics.
self.copy_weights1 = nn.Identity()