深度学习可视化工具——Tensorboard绘制损失和精度曲线

1. 安装tensorboard：
   Cmd命令：pip install tensorboardX

2. 在训练中导入：

   from tensorboardX import SummaryWriter

3. 在程序中调用：

   writer = SummaryWriter(‘log’)

保存要做图的所有信息，在目录下建立一个log文件夹存放。
4. 绘制loss曲线：

writer.add_scalar('Train/Loss', loss.data[0], niter)

在训练过程中，依次写入：图像名称；loss数值；n_interation。
5. 绘制准确度曲线：

writer.add_scalar('Test/Accu', correct/total, niter)

在测试的循环中，写入预测的准确度即可。
6. 加载图像文件：
Cmd命令：cd 到log文件夹同级目录，然后tensorboard –logdir=log，回车
产生一个网页地址，copy到谷歌浏览器打开。（不知为何无法打开，解决办法，直接谷歌浏览器输入：http://localhost:6006，问题解决。）