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(只要你喜欢);

你可能感兴趣的:(conda,装tensorboardx)