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