Torchinfo-viz: see your PyTorch model as an interactive graph

torchinfo-viz turns any PyTorch model into an interactive view: you can see each layer, how layers connect (including skip connections), the tensor shapes, and which layers hold the most parameters. pip install torchinfo-viz built on torchinfo (MIT). feedback welcome! Github: torchinfo-viz