Skip to content

GitLab

Commit b45a91ba authored by Sachin Kadyan's avatar Sachin Kadyan
Browse files

Limit the MSA distillation clusters to 1 in seq mode

parent 3f592307
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
openfold/config.py
View file @ b45a91ba
...@@ -156,9 +156,11 @@ def model_config( ...@@ -156,9 +156,11 @@ def model_config(
elif name == "seqemb_initial_training": elif name == "seqemb_initial_training":
c.data.train.max_msa_clusters = 1 c.data.train.max_msa_clusters = 1
c.data.eval.max_msa_clusters = 1 c.data.eval.max_msa_clusters = 1
c.data.train.max_distillation_msa_clusters = 1
elif name == "seqemb_finetuning": elif name == "seqemb_finetuning":
c.data.train.max_msa_clusters = 1 c.data.train.max_msa_clusters = 1
c.data.eval.max_msa_clusters = 1 c.data.eval.max_msa_clusters = 1
c.data.train.max_distillation_msa_clusters = 1
c.data.train.crop_size = 384 c.data.train.crop_size = 384
c.loss.violation.weight = 1. c.loss.violation.weight = 1.
c.loss.experimentally_resolved.weight = 0.01 c.loss.experimentally_resolved.weight = 0.01
......
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