解决tensorboard每次都是从step3开始,而且不能显示每一步的操作

碰到一个问题,在使用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开始了。而且每一步都能显示出来了

 

  • 8
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

铁蛋阳阳

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值