search

szagoruyko / pytorchviz

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

updated to latest pytorch version; added "trace_and_dot" wrapper #37

Open afqueiruga opened 4 years ago

szagoruyko commented 4 years ago

hmm does not work me on pytorch nightly due to https://github.com/pytorch/pytorch/pull/29149 , it seems to be private now?

afqueiruga commented 4 years ago

Ah, darn. I only updated it to work with the latest release 1.3.1, which is what's on PyPI currently.