"...internal/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "b2aa2317041ed79d390874a8cf89d48452779ddf"
Commit 1d68bde1 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Remove vestigial comment

parent f3f224a9
......@@ -137,6 +137,8 @@ class EvoformerBlock(nn.Module):
):
super(EvoformerBlock, self).__init__()
self._is_extra_msa_stack = _is_extra_msa_stack
self.msa_att_row = MSARowAttentionWithPairBias(
c_m=c_m,
c_z=c_z,
......@@ -212,8 +214,6 @@ class EvoformerBlock(nn.Module):
chunk_size: Optional[int] = None,
_mask_trans: bool = True,
) -> Tuple[torch.Tensor, torch.Tensor]:
#print(torch.cuda.memory_summary())
# DeepMind doesn't mask these transitions in the source, so _mask_trans
# should be disabled to better approximate the exact activations of
# the original.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment