conda 装tensorboardx_【工欲善其事】TensorboardX的使用

“我不喜欢Tensorflow，但这并不妨碍我使用tensorboard”

上一篇文章(https://zhuanlan.zhihu.com/p/39849027)，和大家简单地聊了一下关于如何在训练过程中有序地组织log问题。今天，想和大家简单地谈谈tensorboard的使用。

经过社区的努力，目前PyTorch也可以使用tensorboard了。在训练过程中实时地观察loss/accuracy曲线，并据此来判断网络的训练情况，对每一个DLer来说，想必都是一种极为理想的事情。今天，我们就来简单谈谈tensorboardX的使用。

首先，还是老样子，明确一下我们的需求：1. 实时记录训练过程中的相关数据(训练loss，测试loss，测试Accuracy等等)；

2. 将训练 / 测试相关数据放在一起，方便对比；

一、安装

tensorboardX是针对tensorboard的一个封装，以使得PyTorch用户也能够用到tensorboard。了解到这一点后，我们需要：1)安装tensorboard，2)安装tensorboardX。

安装方法也很简单，使用pip安装，或者在自己的conda虚拟环境中安装都是可以的。

二、使用

tensorboardX的组成比较简单，基本上只有一个类，还有围绕这几个类的一些成员方法。我们主要介绍一下tensorboardX.SummaryWriter以及add_scalar。

在tensorboardX中只有一个类，那就是SummaryWriter，你可以定义许多个SummaryWriter(只要你喜欢)；