Pytorch学习日记——Tensorboard使用(一)

学习视频——B站【小土堆】

代码

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

#writer.add_image()
#y = 2x

for i in range(100):

    writer.add_scalar("y=2x",2*i,i)

writer.close()

代码跑通后,在文件夹里会出现一个新创建的logs文件夹,在pycharm的Terminal中输入

tensorboard --logdir=logs

此时会出现

默认6006端口,如果不想使用此端口,可以按Ctrl+c回退,输入

tensorboard --logdir=logs --port=6007

 指定为6007端口,当然你也可以选择其他端口

此时点击蓝色链接,会跳出浏览器,浏览器中显示的图像就为之前所设计的

Pytorch学习日记——Tensorboard使用(一)_第1张图片

遇到的问题 

过程中:

1.tensorboard模块未安装的情况,参看其他博主输入安装代码即可

2.出现tensorboard包重复的情况,报错代码:

ValueError: Duplicate plugins for name projector

Pytorch学习日记——Tensorboard使用(一)_第2张图片

我的解决办法是在 anaconda3\envs\pytorch\Lib\site-packages 中删除多余的tensorboard文件夹即可,我删除的是这两个:

3.输出图像混乱的情况

一种可能是平滑度(Smoothing),将其调到最低即可;另一种可能是因为logs文件夹中存在未删除的其他图像或错误文件等原因, 删除logs中文件,重新运行一遍代码,输入端口指令,就可以发现函数图像没那么阴间了。

PS:如遇到Pycharm中Terminal为ps啥啥啥而不是(pytorch) C:\.......的,点击

Pytorch学习日记——Tensorboard使用(一)_第3张图片即可 

你可能感兴趣的:(pytorch,学习,深度学习,python)