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

Added training preset for sequence embedding initial training.

- Turn on `seqemb` mode in `data`, `model`, and `globals` config when using `seqemb` training preset.
parent a6a467e0
......@@ -152,6 +152,12 @@ def model_config(
c.model.template.enabled = False
c.model.heads.tm.enabled = True
c.loss.tm.weight = 0.1
# SINGLE SEQUENCE EMBEDDING PRESETS
elif name == "seqemb_initial_training":
c.data.seqemb_mode.enabled = True
c.globals.seqemb_mode_enabled = True
c.model.extra_msa.enabled = False
c.model.evoformer_stack.no_column_attention = True
else:
raise ValueError("Invalid model name")
......
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