TensorBoard 的使用(一)


前言:
  安装完 pytorch 环境后,我们正式学习第一个内容:使用 Tensorboard 来使数据可视化,对应 B站 小土堆 博主中 pytorch 教程的第八节内容,先看视频,再读本文章会比较容易看懂。


1、安装 TensorBoard

  在运行下面主要代码之前,我们要先安装 TensorBoard 这个 python 库。
(这里先停一下,你要先学会安装 python 库,如果不会,请查其他资料学习)

  • 用 pip 下载 python 库;
  • 用 conda 下载 python 库;
  • 使用 PyCharm 软件下载 python 库。
pip install tensorboard

2、代码展示

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("log")

for i in range(100):
    writer.add_scalar("y=2x", 2*i, i)

writer.close()

3、解释代码

  从 torch.utils.tensorboard 中导入 SummaryWriter 库。在当前代码文件(如:a8.py)下的文件夹内创建一个叫做 log 的文件夹,用于存放运行产生的数据文件。这个文件夹的位置与代码文件的位置是相同的。

例如:a8.py 放在 D:\pytorch 中,其中 pytorch 为文件夹的名字,则这个叫 log 的文件夹的路径为: D:\pytorch\log 。

  如果代码是下面这种形式:就会在代码所在的文件夹外创建 log 文件夹。例如:a8.py 放在 D:\pytorch\代码 中,那么 log 文件夹就的路径为: D:\pytorch\log

writer = SummaryWriter("../log")

  将 SummaryWriter( ) 赋给 writer 。i 的值是从 0 到 99 ,总共有 100 个数字。用writer.add_scalar( ) 来存放数值数据,scalar 是标量的意思。里面的参数设置分别为:框名(不同的框名存放不同的数据),y 值,x 值。最后一定要记得用 writer.close( ) 关闭 tensorboard 。运行代码。

4、运行代码

5、打开TensorBoard

  比较方便的方法是:直接在 PyCharm 软件的终端上运行下面代码:

tensorboard --logdir="log"
  第二种方法是在任何终端输入上面的代码,但要在存放 log 文件夹的路径下执行

6、结果展示

  • 24
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

在下_诸葛

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值