Capacités de TAO Skill Bank

Utilisez cette skill quand l'utilisateur demande ce que tao-skill-bank peut faire, demande les capacités du plugin, demande quels workflows d'application ou de données sont disponibles, demande quels modèles sont supportés, ou demande quels modèles sont capables avec AutoML.

Démarrage rapide

Exécutez scripts/list_tao_capabilities.py pour les questions de capacités générales, ou scripts/list_tao_models.py pour les questions de modèles/actions et support AutoML.

Réponses sur les capacités

Pour une réponse générale sur les capacités, exécutez l'assistant fourni :

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_capabilities.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --format text

Utilisez la sortie de l'assistant comme source de vérité pour la réponse au lieu d'énumérer manuellement les capacités à partir de cette skill ou des métadonnées du plugin. Incluez :

• Chaque workflow d'application de haut niveau sous applications/ et ce qu'il peut faire.
• Chaque workflow de données de haut niveau sous data/ et ce qu'il peut faire.
• Plates-formes d'exécution supportées provenant de scripts/list_tao_platforms.py.
• La couverture du workflow fine-tuning/déploiement pour les modèles sous models/ : train, evaluate, inference, export et génération de moteur TensorRT quand ces actions sont présentes dans le manifeste du schéma fourni.
• Support AutoML et la gate du schéma d'entraînement AutoML.

Listes de modèles

Quand l'utilisateur demande quels modèles TAO sont disponibles ou quelles actions un modèle peut exécuter, utilisez le script de liste de modèles fourni au lieu de scanner manuellement les dossiers de modèles :

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_models.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --scope all --format text

La liste des modèles provient de skills/models/schemas.manifest.json.

Listes AutoML

Quand l'utilisateur demande quels modèles sont capables avec AutoML, utilisez le même script de liste de modèles en mode AutoML, ou l'enveloppe de compatibilité :

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_models.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --scope automl --format text

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_automl_support.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --format text

Le support AutoML nécessite que skills/models/<network>/schemas/train.schema.json soit fourni avec le plugin et s'analyse avec succès en JSON. Si ce schéma de dataclass est manquant ou invalide, ne décrivez pas le modèle comme supporté par AutoML.