如何为PyTorch配套使用的TensorBoard页面设置自定义标题

方法1：启动时直接指定标题参数 #

TensorBoard原生支持两个和标题相关的启动参数，直接在启动命令中追加即可实现多实例区分：

• --title：控制TensorBoard页面顶部显示的主标题
• --window_title：控制浏览器标签栏显示的标题

示例命令：

tensorboard --logdir path/to/your/task1/logs --title "仿真任务1：基础参数对照组" --window_title "任务1-TensorBoard"

多开不同TensorBoard实例时，每个实例单独配置上述两个参数即可快速区分。

方法2：单个TensorBoard实例下区分多组任务 #

如果你不想为每个任务单独占用一个端口，可以把不同任务的日志存到同一父目录下的独立子文件夹，子文件夹直接命名为你的任务标记名，目录结构参考如下：

all_simulation_logs/
├── task1_lr0.001_batch32/
├── task2_lr0.01_batch64/
└── task3_lr0.0001_batch16/

启动时指定父目录作为logdir即可：

tensorboard --logdir all_simulation_logs

打开TensorBoard后，侧边栏会自动列出所有子目录对应的任务名，所有指标曲线也会用不同颜色区分不同任务，也可以搭配--title参数给整个页面加总标题，比如--title "XX批次仿真任务全量结果"。

方法3：PyTorch代码层面自定义日志标识 #

如果使用torch.utils.tensorboard.SummaryWriter写日志，初始化时可以通过两个参数自定义日志标识，和方法2配合使用效率更高：

• comment参数：给默认生成的日志文件夹名加自定义后缀，示例代码如下：

from torch.utils.tensorboard import SummaryWriter
# 生成的日志文件夹名会是 runs/[日期时间]_主机名_task1_lr0.001
writer = SummaryWriter(comment="_task1_lr0.001")

• log_dir参数：直接指定日志文件夹的完整路径和名称，完全按照你的标记需求自定义即可。

内容的提问来源于stack exchange，提问作者user1172131