TensorBoard的使用（一）

1. 安装TensorBoard

进入已经安装Pytorch的conda环境

pip3 install tensorboard

2. SummaryWriter类

  • 导入模块

    from torch.utils.tensorboard import SummaryWriter

    按住Ctrl点击SummaryWriter即可插件具体用法 image-20260112104003250

    image-20260112104309787

    log_dir (str)指定日志存储路径，默认为./runs/**CURRENT_DATETIME_HOSTNAME**

    其他不常用的参数:comment (str),purge_step (int),max_queue (int),flush_secs (int),filename_suffix (str). 具体用法可见参数描述

  • 创建实体类writer

    writer = SummaryWriter("logs")

    这会在当前目录下生成一个名为logs的文件夹，用于存放事件文件

3. add_scalar() 添加标量到Summary中

  • 绘制标量数据:writer.add_scalar(tag, scalar_value, global_step)

    • tag：图表的标题（如y=2x）
    • scalar_value: y轴的值
    • global_step：x轴的值

  • 实例: 绘制y=2x的直线

    for i in range(100):
    writer.add_scalar('y=2x',2*i,i)
writer.close()

4. 启动TensorBoard面板

  • 命令行启动: 

    tensorboard --logdir=logs
    • --logdir指定了之前在SummaryWriter中设置的日志文件夹名称。

  • 指定端口（可选）：如果默认端口6006被占用，可使用--port=6007指定其他端口

  • 查看结果：命令运行成功后，点击终端显示的链接（通常是 http://localhost:6006）即可在浏览器中看到可视化的图表。