Record execution graphs of PyTorch neural networks
Project description
torchrecord
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://torchrecord.readthedocs.io/
Installation
Requirements:
- Python3.6+
- PyTorch v1.3 or greater (the
cpu
version) - The Graphviz library and
graphviz
python package.
Install this package:
$ pip install torchrecord
Acknowledgements
This is inspired from szagoruyko/pytorchviz
. This package
differs from pytorchviz
as it provides rendering at multiple depths.
Note that for rendering a network during training, you can use TensorBoard and
torch.utils.tensorboard.SummaryWriter.add_graph
,
which records and renders to a protobuf
in a single step. The intended usage of torchrecord
is for
presentation purposes.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for torchrecorder-1.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 58f0a35f17c4dbea37cd6586b2b3953f8795d7b7b410e1d4fa8e013806b16008 |
|
MD5 | f5c20fb75a8ca8882930ad354e824e42 |
|
BLAKE2b-256 | 3de867135cd485dcee816c304fcf3965b2ae0b5f56d741bd096f0576be7d1d77 |