Skip to content

GitLab

"scripts/git@developer.sourcefind.cn:OpenDAS/openfold.git" did not exist on "13f8f163f68e2ee46e08b7bbbf4fcd7db437e7f2"
Commit 1d68bde1 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Remove vestigial comment

parent f3f224a9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/model/evoformer.py
View file @ 1d68bde1
...@@ -137,6 +137,8 @@ class EvoformerBlock(nn.Module): ...@@ -137,6 +137,8 @@ class EvoformerBlock(nn.Module):
): ):
super(EvoformerBlock, self).__init__() super(EvoformerBlock, self).__init__()
self._is_extra_msa_stack = _is_extra_msa_stack
self.msa_att_row = MSARowAttentionWithPairBias( self.msa_att_row = MSARowAttentionWithPairBias(
c_m=c_m, c_m=c_m,
c_z=c_z, c_z=c_z,
...@@ -212,8 +214,6 @@ class EvoformerBlock(nn.Module): ...@@ -212,8 +214,6 @@ class EvoformerBlock(nn.Module):
chunk_size: Optional[int] = None, chunk_size: Optional[int] = None,
_mask_trans: bool = True, _mask_trans: bool = True,
) -> Tuple[torch.Tensor, torch.Tensor]: ) -> Tuple[torch.Tensor, torch.Tensor]:
#print(torch.cuda.memory_summary())
# DeepMind doesn't mask these transitions in the source, so _mask_trans # DeepMind doesn't mask these transitions in the source, so _mask_trans
# should be disabled to better approximate the exact activations of # should be disabled to better approximate the exact activations of
# the original. # 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