欢迎光临散文网 会员登陆 & 注册

05-03-TensorBoard使用(一)

2022-08-13 10:46 作者:冰糖炖子瑜  | 我要投稿

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 坐标


05-03-TensorBoard使用(一)的评论 (共 条)

分享到微博请遵守国家法律