(P) Package de visualisation Pytorch interactive qui fonctionne dans les ordinateurs portables avec 1 ligne de code

Je travaille sur un package open source "torchvista" Cela vous aide à visualiser la passe avant de votre modèle Pytorch en tant que graphique interactif dans des ordinateurs portables en ligne comme Jupyter, Colab et Kaggle.

Certaines des principales fonctionnalités que je voulais ajouter qui manquaient dans les autres outils que j’ai recherchés étaient

1. Visualisation interactive: y compris l’exploration modulaire des modules imbriqués (en effondrer et en expansion les modules pour masquer / révéler les détails), faire glisser et zoomer
2. offrant une vue claire des formes de divers tenseurs qui traversent le graphique
3. Tolérance aux erreurs: produire un graphique partiel même s’il y a des échecs comme les décalages de forme du tenseur, ce qui facilite le débogage des problèmes pendant que vous construisez des modèles
4. Prise en charge des ordinateurs portables: capacité à s’exécuter dans des ordinateurs portables en ligne comme Jupyter et Colab

Voici le GitHub Repo avec des instructions simples pour l’utiliser. Et voici une procédure pas à pas Google Colab Notebook pour le voir en action (vous devez être connecté à Google pour voir les sorties).

Et voici quelques démos interactives que j’ai faites que vous pouvez voir dans le navigateur:

• MobileVitCollapsed
• Modèle linéaire
• Liste complète des démos

J’adorerais entendre vos commentaires!

Merci!