Pytorch Tensorboard可视化工具

众所周知，好多模型训练的时候不能够看到其训练参数的变化，比如loss曲线，准确率等等，而tensorboard可以有效的解决这个问题，让你的训练过程可视化。以下就是关于tensorboard的使用方法。

1. 安装所需要的库

由于tensorboard原本是在tensorflow里面用的，所以需要装一个tensorflow。tensorflow会自带一个tensorboard。

pip install tensorboard

2.导入tensorboard模块

加在头文件下边

from torch.utils.tensorboard import SummaryWriter

3.创建一个tensorboard问文件

log_dir : tensorboard文件的存放路径
flush_secs：表示每次写入tensorboard文件的时间间隔，30表示半分钟写入一次数据。

writer = SummaryWriter(log_dir='logs',flush_secs=30)