pytorch中tensorboard的使用

保护我方vivian

于 2021-02-04 23:04:59 发布

阅读量216

点赞数

CC 4.0 BY-SA版权

分类专栏：有用的coding 文章标签：深度学习 pytorch tensorflow 神经网络

本文链接：https://blog.csdn.net/qq_42138927/article/details/113665888

有用的coding 专栏收录该内容

2 篇文章

订阅专栏

这篇博客介绍了如何利用TensorBoard来记录和展示机器学习模型的训练过程。通过`SummaryWriter`，作者在每个迭代步长(i)处添加了'y=x'的标量数据，并将其保存在'./log'目录下。最后，通过在终端运行`tensorboard --logdir=./log --port=6007`启动TensorBoard服务，可以可视化这些数据。

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

from tensorboardX import SummaryWriter
writer = SummaryWriter('./log')

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

writer.close()

终端输入

tensorboard --logdir=./log --port=6007

在这里插入图片描述