Reduce max_factor to 0.01

This commit is contained in:
Daniel Povey 2022-03-11 14:41:55 +08:00
parent ab9a17413a
commit 137eae0b95

View File

@ -451,7 +451,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.02, max_factor: float = 0.01,
min_abs: float = 0.5): min_abs: float = 0.5):
super(DerivBalancer, self).__init__() super(DerivBalancer, self).__init__()
self.channel_dim = channel_dim self.channel_dim = channel_dim