search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Migration to torchviz2 #86

Open leo-ware opened 2 months ago

leo-ware commented 2 months ago

As was mentioned in #83, this package is breaking with python 3.12, and it appears to be abandoned by the original maintainer. I've fixed the issue and republished the package as torchviz2.

So, just run: pip install torchviz2

And then, this will work: import torchviz

johndpope commented 1 month ago

thanks again for doing this. when i have time - i'll look at enhancements. shouldn't take long. just use chatgpt / claude.