PXDesign / model /pxdesign /runner /presets.py
anzhi2710gmailcom's picture
Upload folder using huggingface_hub (part 2)
4d20b62 verified
Raw
History Blame Contribute Delete
889 Bytes
# ---------------------------------------------------------------------------
# Preset configurations for the unified pipeline entry
# ---------------------------------------------------------------------------
PRESETS = {
"preview": {
"N_max_runs": 1,
"target_template_rmsd_thres": 2.0,
"N_sample": 100,
"N_step": 400,
"return_topk": 5,
"extended_w_af2": 0.5,
"extended_w_ptx": 0.5,
"early_stop": True,
"min_early_stop_rounds": 0,
"min_early_stop_successes": 1,
},
"extended": {
"N_max_runs": 1,
"target_template_rmsd_thres": 2.0,
"N_sample": 500,
"N_step": 400,
"return_topk": 5,
"extended_w_af2": 0.5,
"extended_w_ptx": 0.5,
"early_stop": True,
"min_early_stop_rounds": 0,
"min_early_stop_successes": 1,
},
}