Change eps range from -3..3 to -2 .. 2

This commit is contained in:
Daniel Povey 2022-12-24 00:09:39 +08:00
parent 72420eef10
commit 2b50ce2247

View File

@ -466,8 +466,8 @@ class BasicNorm(torch.nn.Module):
channel_dim: int = -1, # CAUTION: see documentation.
eps: float = 0.25,
learn_eps: bool = True,
eps_min: float = -3.0,
eps_max: float = 3.0,
eps_min: float = -2.0,
eps_max: float = 2.0,
) -> None:
super(BasicNorm, self).__init__()
self.num_channels = num_channels