东方耀AI技术分享

标题: pytorch模型可视化：pytorchviz [打印本页]

作者: 东方耀 时间: 2020-5-15 11:02
标题: 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（并不总是有效）

作者: 东方耀 时间: 2020-6-18 16:54
import torchscope
torchscope.scope(model, (3, 224, 224), device="cuda")

欢迎光临东方耀AI技术分享 (http://www.ai111.vip/)