Reduce min_prob of ActivationBalancer from 0.1 to 0.05.
This commit is contained in:
parent
c5cb52fed1
commit
476fb9e9f3
@ -482,7 +482,7 @@ class ActivationBalancer(torch.nn.Module):
|
|||||||
scale_gain_factor: float = 0.02,
|
scale_gain_factor: float = 0.02,
|
||||||
min_abs: float = 0.2,
|
min_abs: float = 0.2,
|
||||||
max_abs: float = 100.0,
|
max_abs: float = 100.0,
|
||||||
min_prob: float = 0.1,
|
min_prob: float = 0.05,
|
||||||
):
|
):
|
||||||
super(ActivationBalancer, self).__init__()
|
super(ActivationBalancer, self).__init__()
|
||||||
self.num_channels = num_channels
|
self.num_channels = num_channels
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user