AIKraft/AI/Войти
AIKraft/Skills/TAO обучение NVDINOv2

TAO обучение NVDINOv2

NVDINOv2 for self-supervised visual representation learning. Trains vision transformers via self-distillation (teacher-student) without labels and produces general-purpose visual features. Use when training, exporting, or running inference for a TAO NVDINOv2 backbone. Trigger phrases include

nvidia official Дизайн

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

NVDINOv2 — это навык для самообучаемого изучения визуальных представлений, обучающий vision-трансформеры через самодистилляцию (учитель-студент) без использования размеченных данных. Он производит признаки общего назначения и поддерживает обучение, экспорт и инференс бэкбона TAO NVDINOv2.

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

  • Обучение модели с использованием предварительно подготовленных весов ViT через параметр train.pretrained_model_path.
  • Запуск автоматического поиска гиперпараметров (AutoML) для этапа обучения по умолчанию.
  • Генерация TensorRT-движка для развертывания модели с использованием шаблонов из папки references/.
  • Инференс и экспорт модели в рамках потока действий, не связанных с обучением.

Как работает

  1. Убедиться, что сессия инициализирована плагином TAO skill bank; если нет, запустить навык tao-setup.
  2. Прочитать references/skill_info.yaml для определения политики AutoML (по умолчанию automl_policy: on).
  3. При запросе на обучение проверить наличие schemas/train.schema.json и references/spec_template_train.yaml.
  4. Если AutoML включен и файлы присутствуют, направить действие обучения через tao-skill-bank:tao-run-automl.
  5. При отключении AutoML или отсутствии схем выполнить прямое обучение модели или сообщить об ограничении.
  6. Для развертывания прочитать references/tao-deploy-nvdinov2.md и использовать шаблоны spec_template_deploy_*.yaml.

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

Обучи NVDINOv2 с использованием AutoML и предустановленных весов ViT.
Запусти генерацию TensorRT-движка для NVDINOv2.
Проведи инференс модели NVDINOv2 без использования AutoML.

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

  • Инициализация сессии через навык tao-setup (preflight, credentials, cross-skill discovery).
  • Наличие файлов schemas/train.schema.json и references/spec_template_train.yaml для работы AutoML.
  • Чтение файла references/tao-deploy-nvdinov2.md перед действиями TAO Deploy TensorRT.

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

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

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

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