(self, conv)
| 247 | continue |
| 248 | |
| 249 | def _init_conv(self, conv): |
| 250 | nn.init.kaiming_uniform_( |
| 251 | conv.weight, a=0, mode='fan_in', nonlinearity='relu') |
| 252 | if conv.bias is not None: |
| 253 | nn.init.constant_(conv.bias, 0) |
| 254 | |
| 255 | def _init_norm(self, norm): |
| 256 | if norm.weight is not None: |