Lightlines
By nvidia

tao-finetune-cosmos-reason

nvidia1.7k installs

Fine-tune Cosmos3 on your own video datasets.

What it does

Orchestrates Cosmos3 fine-tuning and evaluation to produce validated checkpoints and task-aware training metrics.

When to use it

Use for Cosmos3-Nano or Cosmos3-Edge training, evaluation, export, or inference workflows.

How to use it

Give it model checkpoints, video datasets, and training settings; it collects missing choices and seeks launch approval before producing checkpoints and metrics in user-owned directories.

What you provide

  • Your files

Uses

Latest review

What you need · 5

Requires Python 3.11 or newer.

Requires PyYAML in the Python environment.


Access · 2

This skill

Hugging Face

Read

Resolves Hugging Face model revisions

SLURM

Execute

Submits GPU training jobs