SummaryWriter 记录和保存训练日志

背景

SummaryWriter 是 PyTorch 中用于记录训练日志的工具，通常与 ​​TensorBoard​​ 配合使用。如果你看到类似 omnisky 的文件（可能是 events.out.tfevents.xxxxxx.omnisky），说明 SummaryWriter 正在向该文件写入日志数据。

安装tensorboard

pip install tensorboard

启动并访问 TensorBoard​

启动tensorBoard；

法1：

# 加载文件夹
tensorboard --logdir=/data/output/board
# -- logdir 指定日志目录（即 args.board_folder 的路径）。

如果日志文件在 /data/lv/expr/TRIS/output/board/events.out.tfevents.xxxxxx.omnisky，TensorBoard 会自动加载。

法2：

# 加载文件
tensorboard --logdir=/data/events.out.tfevents.1744617787.omnisky

访问 TensorBoard​
打开浏览器访问：

http://localhost:6006