A small package to create visualizations of PyTorch execution graphs
Switch branches/tags
Nothing to show
Clone or download
Latest commit b267a9c Aug 23, 2018
Permalink
Failed to load latest commit information.
test add tests Aug 15, 2018
torchviz mark output nodes in green Aug 22, 2018
README.md Update README.md Jan 31, 2018
examples.ipynb mark output nodes in green Aug 22, 2018
setup.py initial commit Jan 30, 2018

README.md

PyTorchViz

A small package to create visualizations of PyTorch execution graphs and traces.

Installation

Install graphviz, e.g.:

brew install graphviz

Install the package itself:

pip install git+https://github.com/szagoruyko/pytorchviz

Usage

screen shot 2018-01-30 at 16 13 01

There are two functions, make_dot to make graphs from any PyTorch functions (requires that at least one input Variable requires_grad), and make_dot_from_trace that uses outputs of torch.jit.trace (does not always work). See examples.ipynb.

Acknowledgements

The script was moved from functional-zoo where it was created with the help of Adam Paszke, Soumith Chintala, Anton Osokin, and uses bits from tensorboard-pytorch.