资源分类
》 资源 》算法》pytorchviz

pytorchviz

2019-10-09 | |  124 |   0 |   0

PyTorchViz

A small package to create visualizations of PyTorch execution graphs and traces.

Installation

Install graphviz, e.g.:

brew install graphviz

Install the package itself:

pip install torchviz

Usage

1570590106255845.png

There are two functions, make_dot to make graphs from any PyTorch functions (requires that at least one input Variable requires_grad), and make_dot_from_trace that uses outputs of torch.jit.trace (does not always work). See examples.ipynb.

Acknowledgements

The script was moved from functional-zoo where it was created with the help of Adam Paszke, Soumith Chintala, Anton Osokin, and uses bits from tensorboard-pytorch.


上一篇：generative_zoo

下一篇：cogitare

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...