PyTorch中如何使用tensorboard可视化

深度使用过tensorflow的同学可能都使用过tensorboard，因为tensorboard这一高级的可视化的工具，很多人对tensorflow也爱不释手，目前除了tensorflow之外还没有哪个深度学习库开发出了一套完美的可视化工具，这也是tensorflow流行的原因之一，如果不使用tensorboard，你想可视化训练流程，那么你只能自己保存变量，自己画曲线。

所以有很多使用别的深度学习框架的人在研究如何将tensorboard移植到他们的框架中来，当然也有很多成功的例子，不然我也不会写这篇文章了，下面我们就来讲几种目前流行的方法。

1.使用Crayon

Crayon是一个支持任何语言使用tensorboard的框架，它的说明文档访问下面的网址，目前他只支持Python和Lua，而且安装过程比较麻烦，需要docker，不推荐使用此方法。

2.使用tensorboard_logger

tensorboard_logger是由TeamHG-Memex开发的使用tensorboard的库，可以访问文档界面，安装也略微有点繁琐，需要安装tensorflow和他们开发的tensorboard_logger，安装完成之后按照文档的使用说明就可以使用tensorboard了。

3.导入一个脚本实现tensorboard

这个办法是我认为最简单的办法，也是我目前使用的办法，只需要安装cpu版的tensorflow，通过pip install tensorflow就能够很快安装上，然后只需要复制这个网址里面的代码到你的项目文件目录，新建一个logger.py的文件，将代码复制进去就ok了。

然后在你的python文件里面输入from logger import Logger，然后在训练之前定义好想存放tensorboard文件的文件夹，logger = Logger('./logs')这里可以使用任何文件夹存放tensorboard文件。

然后在训练过程中可以通过下面的方式记录想要记录的变量

# (1) Log the scalar values
info = {
    'loss': loss.data[0],
    'accuracy': accuracy.data[0]
}

for tag, value in info.items():
    logger.scalar_summary(tag, value, step)

# (2) Log values and gradients of the parameters (histogram)
for tag, value in model.named_parameters():
    tag = tag.replace('.', '/')
    logger.histo_summary(tag, to_np(value), step)
    logger.histo_summary(tag+'/grad', to_np(value.grad), step)

# (3) Log the images
info = {
    'images': to_np(img.view(-1, 28, 28)[:10])
}

for tag, images in info.items():
    logger.image_summary(tag, images, step)

这样我们就将我们需要的变量放进了tensorborad中，然后我们在当前目录下输入tensorbard --logdir='./logs'，这里需要输入自己的文件夹名称，我的文件夹之前定义为了logs，然后你就能够看到下面的界面

Paste_Image.png

在浏览器中输入http://0.0.0.0:6006/，你就能够进到tensorboard界面了，就像下面这样

2.png

1.png

3.png

这样我们就能够成功地在PyTorch中使用tensorboard可视化了，是不是很方便呢。

本文参考自yunjey's github

完整代码已经上传到了github上

欢迎查看我的知乎专栏，深度炼丹

欢迎访问我的博客