Torchviz

Torchviz est une bibliothèque Python qui génère des représentations visuelles (graphes de calcul) des modèles PyTorch en exploitant le graphe de calcul dynamique (dynamic computation graph) créé lors de l’exécution d’un forward pass. Elle s’appuie sur Graphviz pour produire des diagrammes montrant les opérations et les tenseurs.

Niveau

intermediaire

Vous cherchez un autre mot ? Essayez la recherche rapide ci-dessous :

Partager cette page