kermt-finetune

Finetune pretrained KERMT encoder models on labeled CSV datasets.

413|62|Updated Jun 23, 2026
One-click install
npx skills add https://github.com/NVIDIA-BioNeMo/bionemo-agent-toolkit --skill kermt-finetune
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: kermt-finetune
Source: https://github.com/NVIDIA-BioNeMo/bionemo-agent-toolkit/tree/main/plugins/bionemo-agent-toolkit/skills/kermt-finetune
Command: npx skills add https://github.com/NVIDIA-BioNeMo/bionemo-agent-toolkit --skill kermt-finetune

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires docker, nvidia-container-toolkit, and includes scripts (resource) components.

What problem does it solve?

This skill automates the complex, multi-step process of finetuning pretrained KERMT protein encoders on custom labeled datasets, ensuring data integrity and optimal model performance.

Core Features & Use Cases

  • Automated Workflow Orchestration: Handles checkpoint validation, data cleaning, feature preparation, and detached training execution.
  • Flexible Training: Supports regression, classification, and multiclass tasks with automated hyperparameter management and multi-GPU DDP support.
  • Use Case: A researcher has a custom CSV of small molecules with binding affinity labels and needs to adapt the NV-KERMT-70M-v2 model to predict these specific values without manually managing container environments or training scripts.

Quick Start

Use the kermt-finetune skill to train a model on the provided dataset file named binding_data.csv using the default configuration.

Frequently Asked Questions about kermt-finetune

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I finetune a protein encoder model on a custom CSV dataset?▼

To finetune a protein encoder on a custom CSV dataset, you need a labeled molecular dataset and containerized GPU environment. This skill automates data validation, scaffold-balanced splitting, and multi-GPU training orchestration for the finetuning process.

Can I use multi-GPU training for molecular property prediction with Docker?▼

Yes, multi-GPU training for molecular property prediction is supported via Docker. The workflow utilizes multi-GPU DDP support and requires the NVIDIA Container Toolkit alongside CUDA-capable hardware to execute the training scripts.

What data formats are supported for finetuning KERMT models?▼

Finetuning KERMT models requires user-provided labeled CSV datasets containing molecular data such as small molecules with binding affinity labels. The workflow includes automated data validation and scaffold-balanced splitting for these CSV files.

Does finetuning KERMT encoders support classification and regression tasks?▼

Yes, finetuning KERMT encoders supports regression, classification, and multiclass tasks. The workflow provides automated hyperparameter management to adapt the pretrained model to your specific labeled life science dataset.

Do I need CUDA-capable hardware to run kermt-finetune workflows?▼

Yes, CUDA-capable hardware is strictly required to execute kermt-finetune training workflows. You must also install Docker and the NVIDIA Container Toolkit to manage the containerized training environments.

What is the best way to adapt a pretrained encoder for custom life science tasks?▼

The best way to adapt a pretrained encoder for custom life science tasks is using an automated finetuning workflow. It handles checkpoint validation, data cleaning, feature preparation, and detached training execution without manual script management.