Tensorboard不显示全部图像

​
Tensorboard不显示全部图像，我看了b站一个up主的视频链接如下

Dataloader的使用–up主小土堆

但是其中用tensorboard 显示图片我发现显示不全，因为step是跳着的代码如下

import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])

# 准备的测试数据集
test_dataset = torchvision.datasets.CIFAR10(root="./dataset",train=False,transform=dataset_transform,download=True)

#batch_size:每次取多少个数据进行打包
# 参数shuffle true则随机抽取数据
# num_workers=0 是只使用主进程，不使用多进程方式读取数据，window 下使用多进程方式可能会报错
# drops_last是如果dataset内的样本个数不能被batch_size整除，true:舍弃最后不能被整除的数据，false:不舍弃最后的少部分数据（最后读取的一批数据数量比较少）

test_loader = DataLoader(dataset=test_dataset,batch_size=64,shuffle=False,num_workers=0,drop_last=False)

# 测试数据集中第一张图片
img,target = test_dataset[0]
print(img.shape)
print(target)
print(len(test_dataset))
step = 0
writer = SummaryWriter("dataloader_logs")
for data in test_loader:
    images, targets = data
    # print(imgs.shape)
    # print(targets)
    writer.add_images("test_data", images, step)
    print(step)
    step += 1

writer.close()

然后显示的图片如下
​可以发现step从3 开始，然后拉动这个拖动条，第二个step是24非常奇怪

我在命令行输入tensorboard --help 查询到了有个参数叫samples_per_plugin介绍如下：

–samples_per_plugin：
SAMPLES_PER_PLUGIN
An optional comma separated list of plugin_name=num_samples pairs to explicitly specify how many samples to keep per tag for that plugin. For unspecified plugins, TensorBoard randomly downsamples logged summaries to reasonable values to prevent out-of-memory errors for long running jobs. This flag allows fine control over that downsampling. Note that if a plugin is not specified in this list, a plugin-specific default number of samples will be enforced. (for example, 10 for images, 500 for histograms, and 1000 for scalars). Most users should not need to set this flag.

大概意思就是说没有指定这个参数的话，就会默认对你显示的样本进行采样，而不是显示所有的样本
所以如果要显示所有图片需要设置这个参数，我的命令如下

tensorboard --logdir=dataloader_logs --port=6007 --samples_per_plugin=images=10000

这里的10000我测试了应该是你想显示多少步，而不是显示多少张图片，当我设置了20的时候，他抽取了20个step的状态显示，如果你要显示全部的step只需要将这个数设置为大于step最终值就行。（错了勿喷）

结果：
现在可以显示所有的step了
step1：

step2：