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.