(self)
| 59 | self._init_weights() |
| 60 | |
| 61 | def _init_weights(self): |
| 62 | for p in self.parameters(): |
| 63 | if p.dim() > 1: |
| 64 | nn.init.xavier_uniform_(p) |
| 65 | |
| 66 | def forward(self, src, trg, src_mask, tgt_mask, src_padding_mask, tgt_padding_mask, memory_key_padding_mask): |
| 67 |