search

szagoruyko / pytorchviz

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

Update docs #50

Closed soulitzer closed 3 years ago

soulitzer commented 3 years ago

Update readme's old example and also include a new example featuring the latest update.

Preview readme changes at https://github.com/soulitzer/pytorchviz/tree/update-docs See the update to jupyter notebook here: https://github.com/szagoruyko/pytorchviz/blob/71111ea7687ea6ceb0b74d3dd310abf6eb308111/examples.ipynb

The jupyter notebook update is turning out to be a little tricky. So far my workaround is that I'm not running all the cells (leaving some cells out of date) because 1) the jit examples have been failing since forever, and also 2) for larger graphs I'm finding it hard to replicate the scaling from the original notebook (which is able to fit the entire graph in the view), even with code from older commits. Haven't ruled out factors such as different notebook versions, graphviz version or maybe even screen resolution.

cc @szagoruyko @albanD

szagoruyko commented 3 years ago

thanks so much @soulitzer @albanD !