Change chunk tuning defaults

7d442323 · Gustaf Ahdritz · 4c1cf6e2 · 7d442323
Commit 7d442323 authored Jun 12, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

openfold/model/evoformer.py openfold/model/evoformer.py +2 -2

No files found.
--- a/openfold/model/evoformer.py
+++ b/openfold/model/evoformer.py
@@ -498,7 +498,7 @@ class EvoformerStack(nn.Module):
        inf: float,
        eps: float,
        clear_cache_between_blocks: bool = False, 
-        tune_chunk_size: bool = True,
+        tune_chunk_size: bool = False,
        **kwargs,
    ):
        """
@@ -665,7 +665,7 @@ class ExtraMSAStack(nn.Module):
        ckpt: bool,
        clear_cache_between_blocks: bool = False,
        chunk_msa_attn: bool = False,
-        tune_chunk_size: bool = True,
+        tune_chunk_size: bool = False,
        **kwargs,
    ):
        super(ExtraMSAStack, self).__init__()