szagoruyko / pytorchviz

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

Related projects

Alternatives and complementary repositories for pytorchviz