一、SummaryWriter类

二、add_scalar()方法与add_histogram()方法

三、 应用

 

一、SummaryWriter类

要在Python脚本中记录要可视化的数据,以eventfile的形式保存到硬盘中。需要用到SummaryWriter类。

 

8.2 TensorBoard使用

 

log_dir:通常为None,会在py文件的同级目录下创建一个runs的文件夹。runs文件夹下有一些文件夹,再下面是eventfile.

comment:如果不设置log_dir,那么会在runs文件夹下的文件夹会带有comment后缀。

filename_suffix:对eventfile的后缀。

 

例1::如果不设置log_dir。

8.2 TensorBoard使用

 

结果:

8.2 TensorBoard使用

会在同级目录下生成runs文件夹。下面的文件夹的后缀为"scalars"。eventfile后缀为“12345678”.

 

例2:设置log_dir

8.2 TensorBoard使用

 

结果:

8.2 TensorBoard使用

 

有了log_dir,就用不到"_scalars"了。

 

 

二、SummaryWriter类的具体方法

1. add_scalar()

8.2 TensorBoard使用

 

功能:记录标量。

tag:图像标签名,图的唯一标识

scalar_value:要记录的标量,理解为y轴

global_step: x轴。

 

 

例:

 

 

 

2. add_scalars()

add_scalar()只能记录一条曲线。但是往往在模型训练的时候,我们想监控训练集和验证集或者测试集的多条曲线的对比。

这时就可以使用add_scalars()。绘制多个曲线。

 

main_tag: 等同于add_scalar()的tag.

tag_scalar_dict:会用字典的形式记录多个曲线。

 

 

 

 

 

 

 

 

 

 

 

 

相关文章: