(self, weight_init_name)
| 49 | self._reset_parameters(weight_init_name) |
| 50 | |
| 51 | def _reset_parameters(self, weight_init_name): |
| 52 | func = WEIGHT_INIT_DICT[weight_init_name] |
| 53 | for p in self.parameters(): |
| 54 | if p.dim() > 1: |
| 55 | func(p) |
| 56 | |
| 57 | def forward(self, src, |
| 58 | mask: Optional[Tensor] = None, |