HyperAIHyperAI

Command Palette

Search for a command to run...

Apprentissage rapide de représentations graphiques avec PyTorch Geometric

Matthias Fey; Jan Eric Lenssen

Résumé

Nous présentons PyTorch Geometric, une bibliothèque dédiée à l'apprentissage profond sur des données d'entrée irrégulièrement structurées telles que des graphes, des nuages de points et des variétés, construite sur la base de PyTorch. Outre les structures de données générales pour les graphes et les méthodes de traitement associées, elle intègre une variété de méthodes récemment publiées dans les domaines de l'apprentissage relationnel et du traitement de données 3D. PyTorch Geometric atteint un débit élevé de données en exploitant l'accélération GPU creuse, en fournissant des noyaux CUDA dédiés et en introduisant une gestion efficace des mini-lots pour des exemples d'entrée de tailles différentes. Dans ce travail, nous présentons la bibliothèque en détail et effectuons une étude comparative exhaustive des méthodes implémentées dans des scénarios d'évaluation homogènes.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp