Initialize a Linear or Convolutional layer.
(layer)
| 16 | |
| 17 | |
| 18 | def init_layer(layer): |
| 19 | """Initialize a Linear or Convolutional layer.""" |
| 20 | nn.init.xavier_uniform_(layer.weight) |
| 21 | |
| 22 | if hasattr(layer, "bias"): |
| 23 | if layer.bias is not None: |
| 24 | layer.bias.data.fill_(0.0) |
| 25 | |
| 26 | |
| 27 | def init_bn(bn): |
no outgoing calls
no test coverage detected