Venus Fine-Tune Workflow
Overview
Orchestrates training hub tools. Expect CSV columns such as aa_seq / sequence + label. GPU recommended; training can be long — set clear success criteria (metrics file / checkpoint exists).
Project Tools (VenusFactory2)
| Tool | Args | Purpose |
|---|---|---|
| generate_training_config | csv_file or dataset_path; optional valid/test CSV; user_requirements; output_name |
Build training JSON |
| train_protein_model | config_path |
Run training; auto-registers under ckpt/user_trained/<model_id>/ |
| register_trained_model | config_path; optional model_id / output_dir / model_path |
Explicit re-register (usually unnecessary after train) |
| list_trained_models | — | List cross-session registered models |
| protein_model_predict | config_path or model_id + sequence or csv_file |
Inference |
| agent_generated_code | task_description, input_files, output_dir |
Splits, metrics plots, CSV cleanup (Expert only; not MCP) |
Mode: local only. Online mode disables these tools (and VenusMine / FoldSeek discovery). Science Agent reaches train tools via MCP; Expert via the LangGraph tool hub.
Workflow
- Validate/split data with
agent_generated_codeif needed (70/15/15) — Expert path. generate_training_configwith explicituser_requirements(model, epochs, LR, QLoRA…).train_protein_modelon the returned config path (dependency:step_N:file_path). On success, useregistered_model_id/registered_config_pathfrom the result.protein_model_predictwith the registeredmodel_id(or config path) for hold-out or new sequences — works across sessions.- Figure: loss/accuracy curves from training metrics (dpi≥300).
When NOT to use
- Unlabeled mutagenesis ranking →
zero_shot_mutation_workflow - Off-the-shelf solubility/temp heads →
predict_protein_function(protein_property_prediction)
Common mistakes
- Omitting both
csv_fileanddataset_path - Predicting without the same
config_pathused for training - Writing artifacts outside the session
output_dir