Initialize a Linear or Convolutional layer.
(layer)
| 7 | |
| 8 | |
| 9 | def init_layer(layer): |
| 10 | """Initialize a Linear or Convolutional layer. """ |
| 11 | nn.init.xavier_uniform_(layer.weight) |
| 12 | |
| 13 | if hasattr(layer, "bias"): |
| 14 | if layer.bias is not None: |
| 15 | layer.bias.data.fill_(0.0) |
| 16 | |
| 17 | |
| 18 | def init_bn(bn): |
no outgoing calls
no test coverage detected