|
一、SummaryWriter类 二、add_scalar()方法与add_histogram()方法 三、 应用
一、SummaryWriter类要在Python脚本中记录要可视化的数据,以eventfile的形式保存到硬盘中。需要用到SummaryWriter类。
log_dir:通常为None,会在py文件的同级目录下创建一个runs的文件夹。runs文件夹下有一些文件夹,再下面是eventfile. comment:如果不设置log_dir,那么会在runs文件夹下的文件夹会带有comment后缀。 filename_suffix:对eventfile的后缀。
例1::如果不设置log_dir。
结果: 会在同级目录下生成runs文件夹。下面的文件夹的后缀为"scalars"。eventfile后缀为“12345678”.
例2:设置log_dir
结果:
有了log_dir,就用不到"_scalars"了。
二、SummaryWriter类的具体方法1. add_scalar()
功能:记录标量。 tag:图像标签名,图的唯一标识 scalar_value:要记录的标量,理解为y轴 global_step: x轴。
例:
2. add_scalars()add_scalar()只能记录一条曲线。但是往往在模型训练的时候,我们想监控训练集和验证集或者测试集的多条曲线的对比。 这时就可以使用add_scalars()。绘制多个曲线。
main_tag: 等同于add_scalar()的tag. tag_scalar_dict:会用字典的形式记录多个曲线。
|