Passer au contenu

Entraînement de modèles

La plateforme Ultralytics fournit des outils complets pour l'entraînement des modèles YOLO, de l'organisation des expériences à l'exécution de tâches d'entraînement dans le cloud avec le streaming de métriques en temps réel.



Regarder : Premiers pas avec Ultralytics - Entraînement

Aperçu

La section Entraînement vous aide à :

  • Organiser les modèles en projets pour une gestion plus facile
  • Entraîner sur des GPU cloud en un seul clic
  • Surveiller les métriques en temps réel pendant l'entraînement
  • Comparer les performances des modèles entre les expériences
  • Exporter vers plus de 17 formats de déploiement (voir les formats pris en charge)

Présentation de Ultralytics Train

Flux de travail

graph LR
    A[📁 Project] --> B[⚙️ Configure]
    B --> C[🚀 Train]
    C --> D[📈 Monitor]
    D --> E[📦 Export]

    style A fill:#4CAF50,color:#fff
    style B fill:#2196F3,color:#fff
    style C fill:#FF9800,color:#fff
    style D fill:#9C27B0,color:#fff
    style E fill:#00BCD4,color:#fff
ÉtapeDescription
ProjetCréer un espace de travail pour organiser les modèles associés
ConfigurerSélectionnez l'ensemble de données, le modèle de base et les paramètres d'entraînement.
EntraînerExécuter sur des GPU cloud ou votre matériel local
SurveillerVisualiser les courbes de perte et les métriques en temps réel
ExporterConvertir vers plus de 17 formats de déploiement (détails)

Options d'entraînement

La plateforme Ultralytics prend en charge plusieurs approches d'entraînement :

MéthodeDescriptionIdéal pour
Formation CloudFormation sur les GPU UltralyticsPas de GPU local, évolutivité
Formation localeEntraînez-vous localement, transmettez les métriques à la plateformeMatériel existant, confidentialité
Entraînement ColabUtilisez Google avec l'intégration de la plateformeAccès GPU gratuit

Options GPU

GPU disponibles pour l'entraînement dans le cloud sur Ultralytics Cloud :

GPUGénérationVRAMCoût/HeureIdéal pour
RTX 2000 AdaAda16 Go$0.24Petits jeux de données, tests
RTX A4500Ampere20 Go$0.25Jeux de données petits et moyens
RTX 4000 AdaAda20 Go$0.26Jeux de données moyens
RTX A5000Ampere24 Go$0.27Jeux de données moyens
L4Ada24 Go$0.39Optimisé pour l'inférence
A40Ampere48 Go$0.40Tailles de lot plus grandes
RTX 3090Ampere24 Go$0.46Entraînement général
RTX A6000Ampere48 Go$0.49Grands modèles
RTX PRO 4500Blackwell32 Go$0.54Excellent rapport qualité/prix
RTX 4090Ada24 Go$0.59Meilleur rapport performance/prix
RTX 6000 AdaAda48 Go$0.77Entraînement par grands lots
L40SAda48 Go$0.86Entraînement par grands lots
RTX 5090Blackwell32 Go$0.89Dernière génération grand public
L40Ada48 Go$0.99Grands modèles
A100 PCIeAmpere80 Go$1.39Entraînement pour la production
A100 SXMAmpere80 Go$1.49Entraînement pour la production
RTX PRO 6000Blackwell96 Go$1.69Par défaut recommandé
H100 PCIeHopper80 Go$2.39Entraînement haute performance
H100 SXMHopper80 Go$2.69Entraînement le plus rapide
H100 NVLHopper94 Go$3.07Performances maximales
H200 NVLHopper143 Go$3.39Mémoire maximale (Pro+)
H200 SXMHopper141 Go$3.59Performances maximales (Pro+)
B200Blackwell180 Go$4.99Modèles les plus grands (Pro+)

Accès GPU

Les GPU H200 et B200 nécessitent un forfait Pro ou Enterprise. Tous les autres GPU sont disponibles avec tous les forfaits, y compris le forfait Gratuit.

Crédits d'inscription

Les nouveaux comptes reçoivent des crédits d'inscription pour la formation. Consultez la section Facturation pour plus de détails.

Métriques en temps réel

Pendant l'entraînement, consultez les métriques en direct sur trois sous-onglets :

graph LR
    A[Charts] --> B[Loss Curves]
    A --> C[Performance Metrics]
    D[Console] --> E[Live Logs]
    D --> F[Error Detection]
    G[System] --> H[GPU Utilization]
    G --> I[Memory & Temp]

    style A fill:#2196F3,color:#fff
    style D fill:#FF9800,color:#fff
    style G fill:#9C27B0,color:#fff
Sous-ongletMétriques
GraphiquesPerte boîte/classe/DFL, mAP50, mAP50-95, précision, rappel
ConsoleJournaux d'entraînement en direct avec couleurs ANSI et détection des erreurs
SystèmeGPU , mémoire, température, CPU, disque

Points de contrôle automatiques

La plateforme enregistre automatiquement des points de contrôle à chaque époque. Le meilleur modèle ( mAP le plus élevé) et le modèle final sont toujours conservés.

Démarrage rapide

Commencez votre formation sur le cloud en moins d'une minute :

  1. Créer un projet dans la barre latérale
  2. Cliquez sur Nouveau modèle
  3. Sélectionnez un modèle, un ensemble de données et GPU.
  4. Cliquez sur Démarrer l'entraînement
export ULTRALYTICS_API_KEY="YOUR_API_KEY"
yolo train model=yolo26n.pt data=ul://username/datasets/my-dataset \
  epochs=100 project=username/my-project name=exp1
from ultralytics import YOLO

model = YOLO("yolo26n.pt")
model.train(
    data="ul://username/datasets/my-dataset",
    epochs=100,
    project="username/my-project",
    name="exp1",
)

FAQ

Combien de temps dure l'entraînement ?

Le temps de formation dépend de :

  • Taille du jeu de données (nombre d'images)
  • Taille du modèle (n, s, m, l, x)
  • Nombre d'époques
  • Type de GPU sélectionné

Une exécution d'entraînement typique avec 1000 images, YOLO26n, 100 époques sur RTX PRO 6000 prend environ 2 à 3 heures. Les exécutions plus petites (500 images, 50 époques sur RTX 4090) se terminent en moins d'une heure. Voir les exemples de coûts pour des estimations détaillées.

Puis-je entraîner plusieurs modèles simultanément ?

Oui. Le nombre maximal de formations simultanées dans le cloud dépend de votre forfait : le forfait Gratuit autorise 3 formations, le forfait Pro en autorise 10 et le forfait Entreprise est illimité. Pour bénéficier de formations parallèles supplémentaires, utilisez la formation à distance à partir de plusieurs machines.

Que se passe-t-il si l'entraînement échoue ?

Si l'entraînement échoue :

  1. Les points de contrôle sont sauvegardés à chaque époque
  2. Vous pouvez reprendre à partir du dernier point de contrôle
  3. Les crédits ne sont facturés que pour le temps de calcul achevé

Comment choisir le bon GPU ?

ScénarioGPU recommandé
La plupart des emplois dans le domaine de la formationRTX PRO 6000
Grands jeux de données ou tailles de lotH100 SXM ou H200 (Pro+)
Soucieux du budgetRTX 4090


📅 Créé il y a 2 mois ✏️ Mis à jour il y a 10 jours
glenn-jocherRizwanMunawarsergiuwaxmann

Commentaires