SummaryWriter()

SummaryWriter类将条目写入log_dir中的事件文件，以供TensorBoard使用，提供了一个高级API，用于在给定目录中创建事件文件，并向其中添加摘要和事件。该类异步更新文件内容。这允许训练程序调用方法，直接从训练循环向文件添加数据，而不会减慢训练速度；即可以在指定文件夹生成一个事件文件，提供高级接口，该文件可以被TensorBoard解析。

1.导入

from torch.utils.tensorboard import SummaryWriter

2.示例

#第一种：无参数，保存在默认的存储文件夹：runs/日期时间
writer = SummaryWriter()
# folder location: runs/May04_22-14-54_s-MacBook-Pro.local/
 
第二种：保存在指定文件夹中，参数为文件夹名
writer = SummaryWriter("my_experiment")
# folder location: my_experiment
 
第三种：保存在默认文件夹中，提供一个comment参数，功能是在文件夹后加后缀，如runs/日期时间-comment
writer = SummaryWriter(comment="LR_0.1_BATCH_16")
# folder location: runs/May04_22-14-54_s-MacBook-Pro.localLR_0.1_BATCH_16/

3.打开

想要在浏览器中查看可视化的数据，在命令行中开启 tensorboard ：

tensorboard --logdir=事件文件所在的文件夹名

还可以在根目录下输入PowerShell

之后再输入 tensorboard.exe --logdir=./   (老版本)

                    tensorboard.exe --logdir"./"   (新版本)

接着用浏览器打开http://localhost:6006/，即可连接。