(self)
| 596 | self._reset_parameters() |
| 597 | |
| 598 | def _reset_parameters(self): |
| 599 | # init input_proj |
| 600 | for proj in self.input_proj: |
| 601 | nn.init.xavier_uniform_(proj[0].weight, gain=1) |
| 602 | nn.init.constant_(proj[0].bias, 0) |
| 603 | |
| 604 | def prepare_for_dn2(self, targets): |
| 605 | if not self.training: |