Skip to content

GitLab

Commit 0dc316c3 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Update config

parent b066dd55
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/config.py
View file @ 0dc316c3
...@@ -229,7 +229,7 @@ config = mlc.ConfigDict( ...@@ -229,7 +229,7 @@ config = mlc.ConfigDict(
"use_small_bfd": False, "use_small_bfd": False,
"data_loaders": { "data_loaders": {
"batch_size": 1, "batch_size": 1,
"num_workers": 4, "num_workers": 16,
}, },
}, },
}, },
...@@ -343,7 +343,7 @@ config = mlc.ConfigDict( ...@@ -343,7 +343,7 @@ config = mlc.ConfigDict(
"msa_dropout": 0.15, "msa_dropout": 0.15,
"pair_dropout": 0.25, "pair_dropout": 0.25,
"blocks_per_ckpt": blocks_per_ckpt, "blocks_per_ckpt": blocks_per_ckpt,
"clear_cache_between_blocks": True, "clear_cache_between_blocks": False,
"inf": 1e9, "inf": 1e9,
"eps": eps, # 1e-10, "eps": eps, # 1e-10,
}, },
......
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