(self)
| 2653 | self._reset_parameters() |
| 2654 | |
| 2655 | def _reset_parameters(self): |
| 2656 | # init input_proj |
| 2657 | for proj in self.input_proj: |
| 2658 | nn.init.xavier_uniform_(proj[0].weight, gain=1) |
| 2659 | nn.init.constant_(proj[0].bias, 0) |
| 2660 | |
| 2661 | def prepare_for_dn2(self, targets): |
| 2662 | if not self.training: |