05-03-TensorBoard使用(一)

tensorboard使用

当指定log_dir 时,comment失效
log_dir = "./train_log/test_log_dir" writer = SummaryWriter(log_dir=log_dir, comment='_scalars', filename_suffix="12345678")

不指定log_dir 时,在当前文件夹生成runs
writer = SummaryWriter(comment='_scalars', filename_suffix="12345678")

1. add_scalar()
在一个图表中记录一个标量的变化,常用于 Loss 和 Accuracy 曲线的记录。
参数:
tag(string)- 该图的标签,类似于 polt.title。
scalar_value(float or string/blobname)- 用于存储的值,曲线图的 y 坐标
global_step(int)- 曲线图的 x 坐标

2. add_scalars()
在一个图表中记录多个标量的变化,常用于对比,如 trainLoss 和 validLoss 的比较
等
参数:
main_tag(string)- 该图的标签。
tag_scalar_dict(dict)- key 是变量的 tag,value 是变量的值。
global_step(int)- 曲线图的 x 坐标

3. add_histogram()
绘制直方图和多分位数折线图,常用于监测权值及梯度的分布变化情况
tag(string)- 该图的标签,类似于 polt.title。
values(torch.Tensor, numpy.array or string)- 用于绘制直方图的值
global_step(int)- 曲线图的 y 坐标