szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
3,219Updated 7 months ago

Related projects

Alternatives and complementary repositories for pytorchviz