yilin-bao / nnanim Goto Github PK
View Code? Open in Web Editor NEWThis 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.
Home Page: https://github.com/yilin-bao/nnanim
License: MIT License