Published 3 years ago
A small package to create visualizations of PyTorch execution graphs
pip install torchviz
[email protected]
