TensorBoard使用

TensorBoard基础内容

TensorBoard安装

打开anaconda prompt shell，进入已经安装好pytorch的虚拟环境，输入pip install tensorflow
安装好tensorflow后，tensorboard已包含其中

TensorBoard 生成日志文件

导入包

from torch.utils.tensorboard import SummaryWriter
import tensorflow as tf

删除logs_train文件夹下的历史记录

tf.io.gfile.rmtree("./logs_train/")

设置新日志文件位置：存放在文件夹logs_train下(每次训练根据时间新建文件夹)

log_dir = "./logs_train/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")+"/"
writer = SummaryWriter(log_dir)

在每轮训练后，写入日志文件内容（以loss为例）

writer.add_scalar("train_loss", loss.item(), total_train_step)

数据记录完毕，关闭writer

writer.close

可在logs_train文件夹下，看到生成好的日志

日志文件的快速查看

1.在pycharm中，点击工程文件夹，Open In Terminal

2. 在终端输入 tensorboard --logdir=./logs_train
   logs_train是作者日志文件所在文件夹的名称
   
   
3. 双击蓝色网址，即可看到可视化结果

备注：清空tensorboard历史数据：

1. 删除logs_train文件夹下的历史记录

tf.io.gfile.rmtree("./logs_train/")

2. 结束tensorboard进程
   在cmd中输入

taskkill /IM "tensorboard.exe" /F

（备注2尚未经过验证，但是只有1，无法删除之前数据）