Skip to content

tensorboard可视化

约 224 个字 6 行代码 1 张图片 预计阅读时间 1 分钟

在Pytorch中使用Tensorboard可视化训练过程

  • 可视化训练过程
  • 官方 pytorch、tensorboard 可视化
  • 进入PyTorch官网
  • 左侧 Learning PyTorch
  • Visualiazing Models,DATA...
  • tensorboard 可以干什么?
  • 保存网络结构图 GRAPHS
  • 训练集上的损失 loss、learning_rate、验证集上的 accuracy SCALARS
  • 保存权重数值分布 HISTOGRAMS
  • 预测图片信息 IMAGES

关于启动

场景描述:让一个TensorBoard实例持续运行,监视所有实验

(1)一次性启动长期运行的TensorBoard:

Python
screen -S tensorboard_permanent
cd /home/student2023/xiehr2023/UnetTSF
tensorboard --logdir=runs --bind_all --reload_interval=5
# Ctrl+A D 分离screen

(2)每次只需运行您的Python脚本,无需管理TensorBoard:

Bash
python module_3.py

(3)一直使用相同的URL查看结果:

Bash
http://localhost:6006

add(1)vscode自动转发端口:

image-20250418183116571

add(2)可以在 python 脚本中加入启动 tensorboard

好处:只需运行脚本,然后在VSCode中打开端口转发,即可访问最新的TensorBoard结果。不再需要单独启动TensorBoard、记住日志路径等

add_circle2025-04-18 15:56:17update2025-04-18 18:35:31