(self, d_msa=256, d_pair=128, d_hidden=32, p_drop=0.15)
| 105 | |
| 106 | class MSA2Pair(nn.Module): |
| 107 | def __init__(self, d_msa=256, d_pair=128, d_hidden=32, p_drop=0.15): |
| 108 | super(MSA2Pair, self).__init__() |
| 109 | self.norm = nn.LayerNorm(d_msa) |
| 110 | self.proj_left = nn.Linear(d_msa, d_hidden) |
| 111 | self.proj_right = nn.Linear(d_msa, d_hidden) |
| 112 | self.proj_out = nn.Linear(d_hidden*d_hidden, d_pair) |
| 113 | |
| 114 | self.reset_parameter() |
| 115 | |
| 116 | def reset_parameter(self): |
| 117 | # normal initialization |
nothing calls this directly
no test coverage detected