pytorch模型可视化：pytorchviz

东方耀

pytorch模型可视化：pytorchviz




不同于keras的可视化，keras.utils.plot_model。pytorch模型可视化比较复杂，此处使用pytorchviz




命令行安装pytorchviz（之前要安装graphviz，这个我keras时已经安装）
查看：http://www.ai111.vip/thread-934-1-1.html


开始执行：
pip3 install git+https://github.com/szagoruyko/pytorchviz


成功显示：
Installing collected packages: graphviz, torchviz
Successfully installed graphviz-0.14 torchviz-0.0.1




from torchviz import make_dot, make_dot_from_trace


print(model)
g=make_dot(model(torch.rand(1,3,112,112).cuda()),params=dict(model.named_parameters()))
g.view()


torch.rand(1,3,112,112).cuda()为伪造的数据，.cuda因为我的模型在gpu上


成功生成了Digraph.gv.pdf文件 ！




主要有两个函数，make_dot可以从任何PyTorch函数（要求至少有一个输入变量requires_grad）中生成图形，
并make_dot_from_trace使用输出torch.jit.trace（并不总是有效）

东方耀

import torchscope
torchscope.scope(model, (3, 224, 224), device="cuda")