domains / families / science / bionemo-kermt-finetune

bionemo-kermt-finetune

separate

Finetune a pretrained KERMT encoder on a labeled CSV. Validate the checkpoint and data, prepare features, and run containerized training. Use a local checkpoint or optionally download a pinned Hugging Face model bundle using HF_TOKEN if configured. Write model bundles, prepared data, logs, and trained models to user-selected host directories.

When to use

• Finetune pretrained KERMT encoder
• Validate checkpoint and data
• Prepare features
• Run containerized training
• Download pinned Hugging Face model bundle
• Multi-GPU data-parallel DDP finetune
• Regression and classification tasks

When not to use

• Inference with already-finetuned checkpoints
• Non-CUDA capable hosts

Skill page

Source