disable clear_cache_between_blocks

guolinke · Dec 19, 2021 · 6fb440c · 6fb440c
1 parent d368763
commit 6fb440c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfold/config.py b/openfold/config.py
@@ -319,7 +319,7 @@ def model_config(name, train=False, low_prec=False):
                     "msa_dropout": 0.15,
                     "pair_dropout": 0.25,
                     "blocks_per_ckpt": blocks_per_ckpt,
-                    "clear_cache_between_blocks": True,
+                    "clear_cache_between_blocks": False,
                     "inf": 1e9,
                     "eps": eps,  # 1e-10,
                 },