不同于tensorflow提供了强大的可视化功能，pytorch可视化一般借助第三方工具，更通用的方法是使用tensorboardX。
github.com/lanpa/tenso…
使用命令：pip install tensorboardX安装后，如下代码可视化模型：
print(model)
writer = SummaryWriter(log_dir="./logs/",comment="myresnet")
with writer:
writer.add_graph(model,input_to_model=torch.rand(1,3,32,32))
更多内容，访问文档：
tensorboardx.readthedocs.io/en/latest/i…
