TensorBoard

在浏览器里查看训练曲线、图像等日志的本地可视化工具

TensorBoard 是 Google 随 TensorFlow 推出的开源训练可视化工具，但 PyTorch 也能通过 torch.utils.tensorboard 直接写日志。训练时程序把损失、奖励、学习率等标量以及图片、直方图写入日志目录，再运行 tensorboard --logdir 就能在浏览器里实时看曲线、对比多次实验。它完全本地运行、不需要账号，所以很多机器人强化学习框架（如 rsl_rl、Isaac Lab 的训练脚本）默认用它记录日志；需要云端协作和实验管理时，常换成 Weights & Biases 或 SwanLab。

例子在 Isaac Lab 里训练四足行走策略时，打开 TensorBoard 观察平均奖励和回合长度曲线是否在上升，判断训练有没有收敛。

也叫

tensorboard

相关

Weights & Biases、SwanLab（国产训练实验追踪工具）、损失函数、rsl_rl、TensorFlow、PyTorch

来源

TensorBoard 官方页面
tensorflow/tensorboard (GitHub)