Skip to content

A small package to record execution graphs of PyTorch neural networks

License

Notifications You must be signed in to change notification settings

ahgamut/torchrecorder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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:

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.

About

A small package to record execution graphs of PyTorch neural networks

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages