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. channel_dim: int = -1, # CAUTION: see documentation.
eps: float = 0.25, eps: float = 0.25,
learn_eps: bool = True, learn_eps: bool = True,
eps_min: float = -3.0, eps_min: float = -2.0,
eps_max: float = 3.0, eps_max: float = 2.0,
) -> None: ) -> None:
super(BasicNorm, self).__init__() super(BasicNorm, self).__init__()
self.num_channels = num_channels self.num_channels = num_channels