【深度之眼】Pytorch框架班第五期-Tensorboard界面使用详解

最新推荐文章于 2025-10-12 21:38:02 发布

原创

最新推荐文章于 2025-10-12 21:38:02 发布 · 584 阅读

4 ·

CC 4.0 BY-SA版权

本文介绍如何利用TensorBoard与PyTorch的SummaryWriter记录和可视化模型训练过程中的数据，包括不同类型的曲线图展示及TensorBoard界面操作指南。

在这里插入图片描述

import numpy as np
# SummaryWriter用来创建一个writer，可以记录我们想要可视化的数据。
from torch.utils.tensorboard import SummaryWriter


writer = SummaryWriter(comment='tensorboard_818')

for x in range(100):
	# 曲线名称，y轴，x轴
    writer.add_scalar('y=2x', x * 2, x)
    writer.