Tensorboard的使用(一)writer.add_scalar(),SummaryWriter()

1、相关知识

环境与版本

    win10系统 torch --> 1.8.0 tensorboard -->2.4.1
writer.add_scalar()

解释:

二、实例展示-绘制y=2x 

1、打开pycharm,选择环境

操作如下图,选择包含pytorch的环境

2、输入以下代码

from torch.utils.tensorboard import SummaryWriter

#生成一个写日志的writer,并将当前的计算图写入日志,生成的文件在当前文件夹下
writer = SummaryWriter(log_dir="runs/flower_experiment")


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

writer.close()
SummaryWriter(log_dir="runs/flower_experiment") 使用SummaryWriter()创建events,自动生成文件夹名称为runs/flower_experiment文件夹及events名称

运行

2、查看tensorboard界面

在pycharm中terminal终端输入

tensorboard --logdir=runs/flower_experiment
logdir = 事件文件所在文件夹名
出现问题:tensorboard 界面没有显示代码描述图像

事件文件夹也生成了,但不知道为什么读不出来

后来发现是标点符号错误

只需做如下修改

对,改为单引号即可

然后重复2-3步骤得到如下结果

3、其他技巧

    修改端口
经验分享 程序员 微信小程序 职场和发展