Glossaire

PyTorch

Découvre PyTorch, le framework d'apprentissage machine flexible et Python-first qui alimente les innovations en matière d'IA comme Ultralytics YOLO . Construis plus intelligemment et plus rapidement dès aujourd'hui !

Entraîne les modèles YOLO simplement
avec Ultralytics HUB

En savoir plus

PyTorch est un cadre d'apprentissage automatique open-source populaire développé principalement par Meta AI. Il fournit une plateforme flexible et intuitive pour la construction et l'entraînement de réseaux neuronaux, ce qui en fait un favori parmi les chercheurs et qui est de plus en plus adopté pour le déploiement en production. Sa conception Python permet une intégration transparente avec le vaste écosystème de science des données Python , abaissant ainsi la barrière à l'entrée pour les développeurs. PyTorch est le cadre fondamental de nombreux modèles de pointe, notamment Ultralytics YOLO pour la détection d'objets.

Caractéristiques principales de PyTorch

PyTorch se distingue par plusieurs caractéristiques essentielles qui contribuent à son utilisation généralisée :

  • Graphes de calcul dynamiques : Contrairement aux frameworks qui nécessitent de définir l'ensemble du graphe de calcul de manière statique avant l'exécution, PyTorch utilise des graphes de calcul dynamiques. Cela signifie que le graphe est construit à la volée au fur et à mesure que les opérations s'exécutent, ce qui offre une plus grande flexibilité pour les architectures complexes et simplifie le processus de débogage. Ce dynamisme est particulièrement bénéfique dans le cadre de la recherche où l'expérimentation est essentielle.
  • Intégration dePython : PyTorch est un outil natif pour les développeurs Python . Il s'intègre en douceur avec des bibliothèques comme NumPy et SciPy, ce qui permet aux utilisateurs de tirer parti d'outils et d'une syntaxe qui leur sont familiers. La courbe d'apprentissage est donc moins abrupte que celle d'autres frameworks.
  • AccélérationGPU : L'entraînement de modèles complexes d'apprentissage profond est intensif en termes de calcul. PyTorch offre une prise en charge robuste de l'accélérationGPU à l'aide de bibliothèques comme CUDA, ce qui accélère considérablement les temps de formation et d'inférence, ce qui est crucial pour les modèles tels que. Ultralytics YOLOv8.

Applications de PyTorch

La polyvalence de PyTorch le rend adapté à un large éventail de tâches d'apprentissage automatique dans divers domaines.

  • Vision par ordinateur : Il est largement utilisé dans le domaine de la vision par ordinateur (VA) pour des tâches telles que la classification d'images, la détection d'objets et la segmentation d'images. Par exemple, les modèlesYOLO 'Ultralytics sont développés et entraînés à l'aide de PyTorch, en tirant parti de ses performances et de sa flexibilité pour réaliser des détections en temps réel dans des applications allant de la conduite autonome aux systèmes de sécurité.
  • Traitement du langage naturel (NLP) : PyTorch est également un outil puissant pour le NLP, permettant le développement de modèles pour la modélisation du langage, la traduction automatique et l'analyse des sentiments. Sa nature dynamique est bien adaptée au traitement des séquences de longueur variable, courantes dans les données textuelles. Par exemple, de nombreux modèles de transformateurs de pointe utilisés pour des tâches telles que la génération de texte sont implémentés dans PyTorch.

PyTorch dans l'écosystème de l'IA

Alors que PyTorch est très populaire, en particulier dans la communauté des chercheurs, d'autres frameworks comme TensorFlow occupent également une place importante dans le paysage de l'IA. Historiquement, l'un des principaux facteurs de différenciation était l'approche dynamique des graphes de PyTorch par rapport aux graphes statiques de TensorFlow, bien que TensorFlow a ensuite introduit des modes d'exécution dynamiques. Les deux frameworks offrent des outils complets pour la construction, l'entraînement et le déploiement de modèles, mais diffèrent dans la conception de l'API et l'orientation de la communauté, comme l'expliquent en détail des comparaisons telles que Explorer les frameworks d'IA de Vision.

PyTorch bénéficie d'une communauté de développeurs importante et active, d'une documentation complète sur le site officiel dePyTorch , et d'outils comme TorchScript pour la transition des modèles vers les environnements de production. Tu peux commencer à utiliser PyTorch avec les modèles Ultralytics en suivant notre guide de démarrage rapide ou former des modèles facilement en utilisant Ultralytics HUB.

Tout lire