Reduce min_abs from 0.5 to 0.2

This commit is contained in:
Daniel Povey 2022-03-11 18:17:49 +08:00
parent 5eafccb369
commit a0d5e2932c

View File

@ -460,7 +460,7 @@ class DerivBalancer(torch.nn.Module):
def __init__(self, channel_dim: int, def __init__(self, channel_dim: int,
threshold: float = 0.05, threshold: float = 0.05,
max_factor: float = 0.01, max_factor: float = 0.01,
min_abs: float = 0.5): min_abs: float = 0.2):
super(DerivBalancer, self).__init__() super(DerivBalancer, self).__init__()
self.channel_dim = channel_dim self.channel_dim = channel_dim
self.threshold = threshold self.threshold = threshold