(self)
| 17 | self._initialize_weights() |
| 18 | |
| 19 | def _initialize_weights(self): |
| 20 | for m in self.modules(): |
| 21 | if isinstance(m, nn.Conv2d): |
| 22 | nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu') |
| 23 | if m.bias is not None: |
| 24 | nn.init.constant_(m.bias, 0) |
| 25 | |
| 26 | def forward(self, x): |
| 27 | res_list = [] |