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
- 在终端输入
tensorboard --logdir=./logs_train
logs_train是作者日志文件所在文件夹的名称
- 双击蓝色网址,即可看到可视化结果
备注:清空tensorboard历史数据:
- 删除logs_train文件夹下的历史记录
tf.io.gfile.rmtree("./logs_train/")
- 结束tensorboard进程
在cmd中输入
taskkill /IM "tensorboard.exe" /F
(备注2尚未经过验证,但是只有1,无法删除之前数据)