学习视频——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端口,当然你也可以选择其他端口
此时点击蓝色链接,会跳出浏览器,浏览器中显示的图像就为之前所设计的
遇到的问题
过程中:
1.tensorboard模块未安装的情况,参看其他博主输入安装代码即可
2.出现tensorboard包重复的情况,报错代码:
ValueError: Duplicate plugins for name projector
我的解决办法是在 anaconda3\envs\pytorch\Lib\site-packages 中删除多余的tensorboard文件夹即可,我删除的是这两个:
3.输出图像混乱的情况
一种可能是平滑度(Smoothing),将其调到最低即可;另一种可能是因为logs文件夹中存在未删除的其他图像或错误文件等原因, 删除logs中文件,重新运行一遍代码,输入端口指令,就可以发现函数图像没那么阴间了。