(self, x)
| 339 | nn.init.kaiming_normal_(m.weight) |
| 340 | |
| 341 | def forward(self, x): |
| 342 | outputs = {} |
| 343 | x = self.stem(x) |
| 344 | if "stem" in self._out_features: |
| 345 | outputs["stem"] = x |
| 346 | for name in self.stage_names: |
| 347 | x = getattr(self, name)(x) |
| 348 | if name in self._out_features: |
| 349 | outputs[name] = x |
| 350 | |
| 351 | return outputs |
| 352 | |
| 353 | def _freeze_stages(self): |
| 354 | if self.frozen_stages >= 0: |
nothing calls this directly
no outgoing calls
no test coverage detected