Change default prob on limit_param_value from 0.2 to 0.6.
This commit is contained in:
parent
2f4df1278d
commit
ff361a7495
@ -893,7 +893,7 @@ class LimitParamValue(torch.autograd.Function):
|
|||||||
|
|
||||||
def limit_param_value(x: Tensor,
|
def limit_param_value(x: Tensor,
|
||||||
min: float, max: float,
|
min: float, max: float,
|
||||||
prob: float = 0.2):
|
prob: float = 0.6):
|
||||||
# You apply this to (typically) an nn.Parameter during training to ensure that its
|
# You apply this to (typically) an nn.Parameter during training to ensure that its
|
||||||
# (elements mostly) stays within a supplied range. This is done by modifying the
|
# (elements mostly) stays within a supplied range. This is done by modifying the
|
||||||
# gradients in backprop.
|
# gradients in backprop.
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user