碰到一个问题,在使用tensorboard时每次只能从step3开始显示,而且有很多步显示不出来
代码如下
import torchvision
from torch import nn
from torch.nn import Sigmoid
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
input=torchvision.datasets.CIFAR10("./dataset",train=False,
transform=torchvision.transforms.ToTensor())
dataloader=DataLoader(input,batch_size=64)
#神经网络
class sigmoid(nn.Module):
def __init__(self):
super(sigmoid,self).__init__()
self.sigmoid1=Sigmoid()
def forward(self,input):
output=self.sigmoid1(input)
return output
sigmoid1=sigmoid()
writer=SummaryWriter("logs")
step=0
for data in dataloader:
imgs,targets=data
writer.add_images("img_input",imgs,global_step=step)
output=sigmoid1(imgs)
writer.add_images("img_sigmoid",output,step)
step+=1
writer.close()
终端运行
显示结果
看吧,一直都是从step3开始,中间很多都消失了,有的说没使用谷歌,但是换了谷歌浏览器还是没啥用
、
、
解决办法就是在终端运行的命令后面再加上一句 --samples_per_plugin=images=1000,如:
原来的命令是
tensorboard --logdir=F:\pythoncode\Projection1\convolution\logs
修改过后的命令
tensorboard --logdir=F:\pythoncode\Projection1\convolution\logs --samples_per_plugin=images=1000
再看修改过后的效果,从step0开始了。而且每一步都能显示出来了