pytorch padding_Pytorch网络可视化

代码地址

pytorch模型的可视化​github.com

pytorch模型的可视化主要有以下几种方法：

1. 使用tensorboard；
2. 使用netron；
3. 使用torchviz。

1. 使用tensorboard可视化

这里使用tensorboard，搭建一个简单的模型来做demo。

# -*-coding:utf-8 -*-

运行代码后会在"./log"路径下生成一个tfevents文件，进入该路径执行命令："tensorboard --logdir=./ " 然后按照提示打开浏览器就可以看到我们搭建的网络结构了，如下图所示，可以双击打开每一个节点查看其内容。

2. 使用netron可视化

首先是保存模型，只需要在上面的代码最后加上一句，就会生成一个pt文件，然后就可以打开netron进行可视化了。

    # 2. 保存成pt文件后进行可视化
    torch.save(modelviz, "./log/modelviz.pt")

netron安装方法：

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple netron

netron可以打开非常多中类型的网络结构，非常推荐使用，结果如下所示。

3. 使用torchviz可视化

from torchviz import make_dot
# .......
# .......

    # 3. 使用graphviz进行可视化
    out = modelviz(sampledata)
    g = make_dot(out)
    g.render('modelviz', view=False)  # 这种方式会生成一个pdf文件

结果如下所示