Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "dcecc47de90a947e78b1e14a4878514b8d7a0111"
Commit 7d442323 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Change chunk tuning defaults

parent 4c1cf6e2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/model/evoformer.py
View file @ 7d442323
...@@ -498,7 +498,7 @@ class EvoformerStack(nn.Module): ...@@ -498,7 +498,7 @@ class EvoformerStack(nn.Module):
inf: float, inf: float,
eps: float, eps: float,
clear_cache_between_blocks: bool = False, clear_cache_between_blocks: bool = False,
tune_chunk_size: bool = True, tune_chunk_size: bool = False,
**kwargs, **kwargs,
): ):
""" """
...@@ -665,7 +665,7 @@ class ExtraMSAStack(nn.Module): ...@@ -665,7 +665,7 @@ class ExtraMSAStack(nn.Module):
ckpt: bool, ckpt: bool,
clear_cache_between_blocks: bool = False, clear_cache_between_blocks: bool = False,
chunk_msa_attn: bool = False, chunk_msa_attn: bool = False,
tune_chunk_size: bool = True, tune_chunk_size: bool = False,
**kwargs, **kwargs,
): ):
super(ExtraMSAStack, self).__init__() super(ExtraMSAStack, self).__init__()
......
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