Update config

0dc316c3 · Gustaf Ahdritz · b066dd55 · 0dc316c3
Commit 0dc316c3 authored Jan 27, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

openfold/config.py openfold/config.py +2 -2

No files found.
--- a/openfold/config.py
+++ b/openfold/config.py
@@ -229,7 +229,7 @@ config = mlc.ConfigDict(
                "use_small_bfd": False,
                "data_loaders": {
                    "batch_size": 1,
-                    "num_workers": 4,
+                    "num_workers": 16,
                },
            },
        },
@@ -343,7 +343,7 @@ config = mlc.ConfigDict(
                "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,
            },