Skip to content

GitLab

Commit 53c03a6a authored by Dingquan Yu's avatar Dingquan Yu
Browse files

update config.py for the development for now

parent 4e58a6a0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
openfold/config.py
View file @ 53c03a6a
...@@ -432,7 +432,7 @@ config = mlc.ConfigDict( ...@@ -432,7 +432,7 @@ config = mlc.ConfigDict(
"use_small_bfd": False, "use_small_bfd": False,
"data_loaders": { "data_loaders": {
"batch_size": 1, "batch_size": 1,
"num_workers": 16, "num_workers": 1,
"pin_memory": True, "pin_memory": True,
}, },
}, },
...@@ -764,7 +764,7 @@ multimer_config_update = mlc.ConfigDict({ ...@@ -764,7 +764,7 @@ multimer_config_update = mlc.ConfigDict({
], ],
"true_msa": [NUM_MSA_SEQ, NUM_RES] "true_msa": [NUM_MSA_SEQ, NUM_RES]
}, },
"max_recycling_iters": 20, # For training, value is 3 "max_recycling_iters": 1, # For training, value is 3
"unsupervised_features": [ "unsupervised_features": [
"aatype", "aatype",
"residue_index", "residue_index",
...@@ -799,7 +799,7 @@ multimer_config_update = mlc.ConfigDict({ ...@@ -799,7 +799,7 @@ multimer_config_update = mlc.ConfigDict({
"train": { "train": {
"max_msa_clusters": 508, "max_msa_clusters": 508,
"max_extra_msa": 2048, "max_extra_msa": 2048,
"crop_size": 640, "crop_size": 32,
"spatial_crop_prob": 0.5, "spatial_crop_prob": 0.5,
"interface_threshold": 10., "interface_threshold": 10.,
"clamp_prob": 1., "clamp_prob": 1.,
......
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