LucaBonfiglioli/nnviz
Torch nn vizualization
This tool helps machine learning engineers and researchers visualize the architecture of PyTorch neural networks. It takes a PyTorch model (either a built-in one, from an installed package, or a local Python file) and generates a clear, comprehensible PDF, PNG, or SVG diagram of its structure. The output shows how different layers and components connect, making it easier to understand complex models.
No commits in the last 6 months. Available on PyPI.
Use this if you need to quickly generate detailed, static diagrams of your PyTorch model architectures for documentation, presentations, or debugging.
Not ideal if you need an interactive visualization tool or are working with deep learning frameworks other than PyTorch.
Stars
57
Forks
—
Language
Python
License
Unlicense
Category
Last pushed
Sep 18, 2023
Commits (30d)
0
Dependencies
7
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/LucaBonfiglioli/nnviz"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
lutzroeder/netron
Visualizer for neural network, deep learning and machine learning models
mert-kurttutan/torchview
torchview: visualize pytorch models
raghakot/keras-vis
Neural network visualization toolkit for keras
janosh/diagrams
Diagrams of concepts in physics/chemistry/ML
dvgodoy/deepreplay
Deep Replay - Generate visualizations as in my "Hyper-parameters in Action!" series!