(self)
| 244 | self.reset_parameters() |
| 245 | |
| 246 | def reset_parameters(self): |
| 247 | |
| 248 | for m in self.parameters(): |
| 249 | if isinstance(m, (nn.Linear, nn.Conv2d)): |
| 250 | nn.init.kaiming_normal_(m.weight) |
| 251 | nn.init.zeros_(m.bias) |
| 252 | |
| 253 | @torch.no_grad() |
| 254 | def load_pretrained(self, state_dict, lookup_22k): |