Record execution graphs of PyTorch neural networks
A small package to record execution graphs of neural networks in PyTorch.
The package uses hooks and the
grad_fn attribute to record information.
This can be used to generate visualizations at different scope depths.
Licensed under MIT License. View documentation at https://pytorchrec.readthedocs.io/
- PyTorch v1.3 or greater (the
- The Graphviz library and
Install this package:
$ pip install torchrec
This is inspired from
szagoruyko/pytorchviz. This package
pytorchviz as it provides rendering at multiple depths.
Note that for rendering a network during training, you can use TensorBoard and
which records and renders to a
protobuf in a single step. The intended usage of
pytorchrec is for
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size torchrec-1.0.1-py3-none-any.whl (12.8 kB)||File type Wheel||Python version py3||Upload date||Hashes View|
|Filename, size torchrec-1.0.1.tar.gz (10.2 kB)||File type Source||Python version None||Upload date||Hashes View|