Record execution graphs of PyTorch neural networks
Project description
torchrecorder
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://torchrecorder.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 torchrecorder
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 torchrecorder
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.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 537ef5b32476756a06446a205d73a4a6d8d164d2101ce87891875f989e9c3311 |
|
MD5 | b754d7f0b9f9378b9cb16db1c70681e7 |
|
BLAKE2b-256 | 37a6ade2011ad5debc5afef7339bb5057140b2440bfdcd346f8188f7c5cc32e8 |