(self)
| 31 | self.reset_parameter() |
| 32 | |
| 33 | def reset_parameter(self): |
| 34 | # initialize weights to normal distrib |
| 35 | self.proj_pair = init_lecun_normal(self.proj_pair) |
| 36 | self.proj_state = init_lecun_normal(self.proj_state) |
| 37 | |
| 38 | # initialize bias to zeros |
| 39 | nn.init.zeros_(self.proj_pair.bias) |
| 40 | nn.init.zeros_(self.proj_state.bias) |
| 41 | |
| 42 | def forward(self, msa, pair, rbf_feat, state): |
| 43 | ''' |
no test coverage detected