(self)
| 379 | |
| 380 | self.reset_parameter() |
| 381 | def reset_parameter(self): |
| 382 | self.proj_state = init_lecun_normal(self.proj_state) |
| 383 | nn.init.zeros_(self.proj_state.bias) |
| 384 | self.proj_state2 = init_lecun_normal(self.proj_state2) |
| 385 | nn.init.zeros_(self.proj_state2.bias) |
| 386 | |
| 387 | def forward(self, seq, msa, msa_full, pair, xyz_in, state, idx, use_checkpoint=False, motif_mask=None): |
| 388 | """ |
no test coverage detected