This library is designed to generate visualizations of neural networks by analyzing PyTorch code. Unlike conventional methods, our approach eliminates the need for large parameter files (pth) or specific input formats. With minimal AI knowledge, users can leverage this tool to create visualizations that aid in understanding complex networks.
python machine-learning deep-neural-networks deep-learning numpy tensor visualization-library visulization nerual-network visualization-tools visualization-toolbox
- Updated
Dec 10, 2023 - Python