PyTorch SummaryWriter TensorBoard 中进行可视化

在 PyTorch 中,SummaryWriter 通常用于在训练过程中记录各种数据,以便在 TensorBoard 中进行可视化。

安装: pip install tensorboard -i

https://mirrors.aliyun.com/pypi/simple/

复制代码
from torch.utils.tensorboard import SummaryWriter

# 创建 SummaryWriter 对象
writer = SummaryWriter('logs')

# 假设在训练过程中,记录损失值
for epoch in range(10):
    loss = 2*epoch  # 这里用简单的示例,实际中是您计算的损失值
    writer.add_scalar('Loss', loss, epoch)

# 关闭 SummaryWriter
writer.close()

writer.add_image(tag, img_tensor, global_step, dataformats)

参数说明:

  • tag(string):数据名称。
  • values(torch.Tensor、numpy.array 或 string/blobname):用来构建直方图的数据。
  • bins(string, 可选):取值有'tensorflow'、'auto'、'fd'等,该参数决定了分桶的方式。
  • walltime(float, 可选):记录发生的时间,默认为 time.time()
  • max_bins(int, 可选):最大分桶数。

可以在终端中运行 tensorboard --logdir=runs/experiment_1 ,然后在浏览器中打开显示的 URL 即可查看可视化结果。

Loss是图的名称,epoch是x轴,Loss是y轴

如果继续绘制,他会持续保存之前的图 ,要清空的话,把logs文件删了就是了

相关推荐
AloneCat201237 分钟前
Qwen2.5简要全流程以及Q&A
深度学习
OceanBase数据库官方博客2 小时前
向量检索新选择:FastGPT + OceanBase,快速构建RAG
人工智能·oceanbase·分布式数据库·向量数据库·rag
ghjhjjjbjibh2 小时前
AI大模型 —— 国产大模型 —— 华为大模型
人工智能·华为
清风序来3 小时前
一,开发环境安装
人工智能·深度学习
樱花穿过千岛湖6 小时前
第六章:Multi-Backend Configuration
人工智能·python·gpt·学习·ai
跳跳糖炒酸奶6 小时前
第十五讲、Isaaclab中在机器人上添加传感器
人工智能·python·算法·ubuntu·机器人
前进的程序员7 小时前
深度学习:人工智能的核心驱动力
人工智能
_一条咸鱼_7 小时前
Python 名称空间与作用域深度剖析(二十七)
人工智能·python·面试
_一条咸鱼_7 小时前
Python之函数对象+函数嵌套(二十六)
人工智能·python·面试
_一条咸鱼_7 小时前
Python 文件操作之修改(二十二)
人工智能·python·面试