(self, num_features, eps=1e-5, momentum=0.1)
| 26 | """ |
| 27 | |
| 28 | def __init__(self, num_features, eps=1e-5, momentum=0.1): |
| 29 | super().__init__() |
| 30 | |
| 31 | self.num_features = num_features |
| 32 | self.eps = eps |
| 33 | self.momentum = momentum |
| 34 | self.weight = nn.Parameter(torch.empty(num_features)) |
| 35 | self.bias = nn.Parameter(torch.empty(num_features)) |
| 36 | self.register_buffer("running_mean", torch.zeros(num_features)) |
| 37 | self.register_buffer("running_var", torch.ones(num_features)) |
| 38 | self.reset_parameters() |
| 39 | |
| 40 | def reset_running_stats(self): |
| 41 | self.running_mean.zero_() |
nothing calls this directly
no test coverage detected