(self, n)
| 24 | nans. |
| 25 | """ |
| 26 | def __init__(self, n): |
| 27 | super(FrozenBatchNorm2d, self).__init__() |
| 28 | self.register_buffer('weight', torch.ones(n)) |
| 29 | self.register_buffer('bias', torch.zeros(n)) |
| 30 | self.register_buffer('running_mean', torch.zeros(n)) |
| 31 | self.register_buffer('running_var', torch.ones(n)) |
| 32 | |
| 33 | def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, |
| 34 | missing_keys, unexpected_keys, error_msgs): |