{"repo":"mert-kurttutan/torchview","free":true,"listed":false,"github":"https://github.com/mert-kurttutan/torchview","clone":"git clone https://github.com/mert-kurttutan/torchview.git","description":"torchview: visualize pytorch models","language":"Python","stars":1074,"topics":["deep-learning","keras","machine-learning","neural-network","pytorch","visualization","python","torchviz"],"license":"MIT","category":"machine-learning","readme_excerpt":"torchview Torchview provides visualization of pytorch models in the form of visual graphs. Visualization includes tensors, modules, torch.functions and info such as input/output shapes. Pytorch version of plot model of keras (and more) Supports PyTorch versions $\\geq$ 1.7. Useful features Installation First, you need to install graphviz, For python interface of graphiz to work, you need to have dot layout command working in your system. If it isn't already installed, I suggest you run the following depeding on your OS, Debian-based Linux distro (e.g. Ubuntu): Windows: macOS see more details here Then, continue with installing torchview using uv (or pip) or if you want via conda or if you want most up-to-date version, install directly from repo How To Use Notebook Examples For more examples, see colab notebooks below, Introduction Notebook: Computer Vision Models: NLP Models: Note: Output graphviz visuals return images with desired sizes. But sometimes, on VSCode, some shapes are being cropped due to large size and svg rendering on by VSCode. To solve this, I suggest you run the following This problem does not occur on other platforms e.g. JupyterLab or Google Colab. Supported Features Almost all the models, RNN, Sequentials, Skip Connection, Hugging Face Models Support for meta tensors - no memory consumption (for very Large models) (pytorch version $\\geq$ 1.13) . Shows operations between tensors (in addition to module calls) Rolling/Unrolling feature. Recursively used module","default_branch":null,"files":null,"tree":[],"storefront":"/r/mert-kurttutan","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/mert-kurttutan/torchview/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}