AIKraft/AI/Войти
AIKraft/Skills/TAO дообучение Cosmos Reason

TAO дообучение Cosmos Reason

Cosmos3-Nano video QA supervised fine-tuning with FSDP parallelism. Use when training or evaluating video question-answering models, fine-tuning Cosmos3-Nano or compatible Cosmos Reason models with SFT/LoRA, or working with Cosmos-RL. Trigger phrases include

nvidia official Другое

Что делает навык

Навык обеспечивает контролируемое дообучение (SFT) и оценку моделей вопросов-ответов по видео (Cosmos3-Nano, Cosmos3-Edge) с использованием параллелизма FSDP. Он управляет полным циклом подготовки данных, выбора бэкенда (Cosmos-RL или Cosmos Framework), конвертации чекпоинтов и запуска обучения или оценки.

Когда применять

  • Дообучение модели Cosmos3-Nano методом SFT или LoRA на видео-датасетах.
  • Оценка качества ответов модели на видео с использованием Cosmos-RL или Cosmos Framework.
  • Автоматическая конвертация чекпоинтов формата cosmos3_omni в совместимый формат qwen3_vl перед обучением.
  • Запуск обучения или микросервисного вывода на кластере SLURM с использованием SQSH-образов.

Как работает

  1. Собрать обязательные входные данные: путь к базовой модели, тип модели (qwen3_vl или cosmos3_omni), режим выборки видео, пути к аннотациям и бэкенд (Cosmos-RL или Cosmos Framework).
  2. Выполнить обязательные проверки (gates): валидацию путей, токенов, инструментов и разрешить автоматический выбор бэкенда или использовать явный.
  3. При выборе cosmos3_omni инициировать автоматическую конвертацию чекпоинта через TAO-интеграцию в бэкенд-образе, сохраняя исходный чекпоинт неизменным.
  4. Проверить целостность данных, разрешить профили видео-рантайма и сгенерировать нативные TOML-спецификации и метаданные.
  5. Запустить обучение или оценку через соответствующий workflow (cosmos_workflow.py или evaluation_workflow.py), передав валидный spec_bundle.
  6. Для оценки через Cosmos Framework выполнить предварительное действие (pre-action) для экспорта чекпоинтов DCP перед запуском.

Примеры запросов агенту

Запусти дообучение Cosmos3-Nano в режиме PEFT с использованием бэкенда Cosmos-RL.
Оцени модель Cosmos3-Edge на видео-датасете с помощью Cosmos Framework.
Подготовь чекпоинт cosmos3_omni для обучения и запусти SFT с параллелизмом FSDP.

Что понадобится

  • Доступ к Hugging Face Hub (токен HF_TOKEN для приватных моделей).
  • Окружение SLURM с поддержкой Pyxis/Enroot и SQSH-образов.
  • Установленные пакеты cosmos-rl или cosmos-framework.

Описание составлено по SKILL.md навыка, сверено 05.10.2026.

Как подключить

1Скачать навык
# возьмите папку навыка «tao-finetune-cosmos-reason» из репозитория: # https://github.com/nvidia/skills
2Положить папку с SKILL.md к навыкам ассистента
# положите папку навыка туда, где ассистент ищет skills: cp -r tao-finetune-cosmos-reason/ ~/.claude/skills/tao-finetune-cosmos-reason/

Навык — папка с файлом SKILL.md (описание + инструкции) и опциональными скриптами. Ассистент (Claude, Claude Code и совместимые) подхватывает его по описанию и следует шагам.