使用pytorchviz进行pytorch执行过程的可视化

Song • 112 次浏览 • 0 个回复 • 2018年02月11日

PyTorchViz

一个小软件包,用于创建PyTorch执行图形和轨迹的可视化。

安装

首先安装graphviz,例如Mac安装:

brew install graphviz

使用pip安装graphviz

pip install git+https://github.com/szagoruyko/pytorchviz

使用方法

首先引入torch以及graphviz

import torch
from torch.autograd import Variable
from torch import nn
from torchviz import make_dot, make_dot_from_trace

然后使用如下代码

model = nn.Sequential()
model.add_module('W0', nn.Linear(8, 16))
model.add_module('tanh', nn.Tanh())
model.add_module('W1', nn.Linear(16, 1))

x = Variable(torch.randn(1,8))
y = model(x)

make_dot(y.mean(), params=dict(model.named_parameters()))

使用pytorchviz进行pytorch执行过程的可视化

主要有两个函数,make_dot可以从任何PyTorch函数(要求至少有一个输入变量requires_grad)中生成图形,并make_dot_from_trace使用输出torch.jit.trace(并不总是有效)。参见examples.ipynb

项目地址:使用pytorchviz进行pytorch执行过程的可视化


原创文章,转载请注明 :使用pytorchviz进行pytorch执行过程的可视化 - pytorch中文网
原文出处: https://ptorch.com/news/128.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! pytorch v0.3.1发布,添加余弦退火方案来调整学习率以及添加random_split/reduce等函数