tensorboard使用

最新推荐文章于 2025-06-10 16:31:31 发布

原创最新推荐文章于 2025-06-10 16:31:31 发布 · 389 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #python

本文介绍了如何使用PyTorch的TensorBoard工具，通过'Scatter'图表记录并可视化训练过程中'y=x'的标量数据，展示了如何设置数据标识、值和全球步长。通过'logs'文件夹展示结果，便于数据分析和模型监控。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logs')

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

writer.close()

官方帮助文档：

相当于Add scalar data to summary.

Args:
    tag (string): Data identifier  # 生成图像的标题
    scalar_value (float or string/blobname): Value to save   # 标量值，相当于Y轴
    global_step (int): Global step value to record   #相当于横坐标