(self, encoder_layer, num_layers,
norm=None, weight_init_name="xavier_uniform")
| 41 | class TransformerEncoder(nn.Module): |
| 42 | |
| 43 | def __init__(self, encoder_layer, num_layers, |
| 44 | norm=None, weight_init_name="xavier_uniform"): |
| 45 | super().__init__() |
| 46 | self.layers = get_clones(encoder_layer, num_layers) |
| 47 | self.num_layers = num_layers |
| 48 | self.norm = norm |
| 49 | self._reset_parameters(weight_init_name) |
| 50 | |
| 51 | def _reset_parameters(self, weight_init_name): |
| 52 | func = WEIGHT_INIT_DICT[weight_init_name] |
no test coverage detected