最近,在“炼丹”的过程中,遇到了tensorboard里的一系列问题,然后按照网上的各种教程尝试......这一过程中收获了很多,所以决定分享一下,如有错误还请及时指正,一起进步。
【方案一】、首先,检查自己的程序是否出错,一定要仔细检查,尤其是关于tensorboard的,这里提供一段范例程序,可以用来检查尝试。
import torchvision
#准备数据集
from torch.utils.tensorboard import SummaryWriter
dataset_transform = torchvision.transforms.Compose([
torchvision.transforms.ToTensor()
])
train_set = torchvision.datasets.CIFAR10(root="./CIFAR10", train=True, transform=dataset_transform, download=True)
test_set = torchvision.datasets.CIFAR10(root="./CIFAR10", train=False, transform=dataset_transform, download=True)
#这里用到CIFAR10数据集,第一次需要下载,默认下载在根目录文件下,之后便可直接使用,download=False
#如何在tensorboard上显示
writer = SummaryWriter("logs") #日志文件保存位置
for i in range(10):
img, target = test_set[i]
writer.add_image("test_set", img, i)
#一定要记得加该行!!!
writer.close()
然后就是如何在tensorboard上显示,有两种方法:
(1)直接在pytorch上使用terminal,敲入
tensorboard --logdir=logs
#如果要更改指定端口,换到其他端口
tensorboard --logdir=logs --port=6007
#该端口可以按照要求进行修改
(2)使用Anaconda Prompt,这里需要将环境配置为pytorch环境,输入:
activate pytorch
然后按照(1)的操作即可。
【方案二】tensorboard版本问题,需要检查版本或者重现安装。(1)在pytorch中直接使用“pip install tensoboard”下载安装资源包;(2)在pytorch中的setting中检查版本号,或升级到最新版本即可。
【方案三】网上大量博主提供的教程
(1)将=改为" ";
(2)浏览器的问题,这种情况下更改默认浏览器即可;
(3)tensorboard --logdir=logs 将这里的logs改为它的绝对路径即可。
以上就是我看到的全部解决办法,如有错误请直接指出,如有帮助可以点个赞。
我们一起进步!