by jsklearner

@[TOC](pytorch 使用tensorboardX可视化网络结构图)

前提条件

安装tensorflow，tensorboardX，pytorch

使用方法

代码

from tensorboardX import SummaryWriter
mynetwork = MyNetWork()   # 定义的网络类名
input_ = torch.randn((1, 3, 224, 224))
writer = SummaryWriter(log_dir='')   
myusbsequenet.train()
writer.add_graph(myusbsequenet, input_)
pred = myusbsequenet(input_)

使用

运行程序后： 生成类似下图中的文件夹，文件夹内有相应文件：

进到该文件夹所在目录： 执行下列命令： 这里需要注意下，有的博客里写的是--logdir=，是有问题的，之前的版本这样写没问题，但是之后就改了，所以遇到问题时可以注意一下这个地方。

tensorboard --logdir .\Dec09_15-34-19_3FWLMWN8W1PCG57\

结果如下图，将图中的网址输入到浏览器即可看到网络结构图：

结果展示：

*** 2019.12.10 希望能帮到你。