conda 装tensorboardx_在pytorch下使用tensorboardX（win10;谷歌浏览器;jupyter notebook）

最新推荐文章于 2024-07-22 10:34:13 发布

易近民

最新推荐文章于 2024-07-22 10:34:13 发布

阅读量727

点赞数

文章标签： conda 装tensorboardx

本文链接：https://blog.csdn.net/weixin_42493060/article/details/111895545

版权

本文介绍了在Windows 10环境下，通过conda安装TensorboardX，并在Jupyter Notebook中配合PyTorch使用Tensorboard进行训练过程的可视化。详细步骤包括conda安装、在Jupyter中启动Notebook、使用SummaryWriter记录训练信息，以及如何通过tensorboard展示loss和accuracy等图表。此外，还列举了如add_scalar、add_scalars、add_histogram等关键函数的使用方法。

摘要由CSDN通过智能技术生成

使用环境：win10 ，在jupyter notebook下运行谷歌浏览器

1、环境安装

使用conda 安装，打开anacond powershell，输入pip install tensorboard ，然后安装pip install tensorflow

2、使用操作

在终端或者环境命令行下，打开程序所在目录，使用shift+右键进入cmd，输入jupyter notebook，进入环境，打开程序。

(1)在程序开头加入

from torch.utils.tensorboard import SummaryWriter，而不是from tensorboardX import SummaryWriter，参考https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html

(2)writer = SummaryWriter('runs/exp1')，runs/exp1表示相关信息存储位置，可自行定义。然后调用相关函数即可。例如想知道训练次数epoch与损失函数之间的关系，调用writer.add_scalar('Train Loss', train_loss / len(train_data), global_step=epoch) 。然后运行一次函数，等待程序运行完毕。

(3)打开runs文件夹的上一级目录，然后使用shift+右键进入cmd，输入tensorboard --logdir=runs ,随后弹出如下

复制网址，然后打开(据说只能使用谷歌浏览器，未经验证)。有可能出现如下提示，打开exp1文件夹，如果文件大小为0，则表示数据未成功写入，重新运行程序即可。或者重新操作第三步

如果进入如下类似页面，则表示成功

指令集补充

(1)add_scalar(tag, scalar_value, global_step=None, walltime=None)

功能：

在一个图表中记录一个标量的变化，常用于 Loss 和 Accuracy 曲线的记录。

参数：tag(string)- 该图的标签，类似于 polt.title。

scalar_value(float or string/blobname)- 用于存储的值，曲线图的 y 坐标

global_step(int)- 曲线图的 x 坐标

walltime(float)- 为 event 文件的文件名设置时间，