Pytorch之可視化

先解決下keras可視化安裝graphviz的問題：

注意安裝順序：

sudo pip3 install graphviz  # python包

sudo apt-get install graphviz   # 軟件本身

sudo pip3 install pydot

sudo pip3 install pydot-ng  # 版本兼容需要,可選

 

1.  使用pytorchviz進行pytorch執行過程的可視化

sudo pip3 install git   # 安裝git
sudo  pip3 install git+https://github.com/szagoruyko/pytorchviz   # github上的包

demo：

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()))

 

詳見： https://github.com/szagoruyko/pytorchviz