ValueError: num_samples should be a positiveinteger value, but got num_samples=0

最新推荐文章于 2024-10-12 14:56:49 发布

原创

最新推荐文章于 2024-10-12 14:56:49 发布 · 993 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习

在使用PyTorch实现GAN时遇到ValueError，源于DataLoader的shuffle参数设置错误。文章解释了如何修正这个问题，即在训练DataLoader中将shuffle设置为False，以避免重复采样。

在跑pytorch-GAN中的代码，出现了“ValueError: num_samples should be a positiveinteger value, but got num_samples=0：问题，

查阅资料，出现的问题的地方：调用DataLoder时注意参数

# Training data loader
dataloader = DataLoader(
    ImageDataset("../../data/%s" % opt.dataset_name, transforms_=transforms_, unaligned=True),
    batch_size=opt.batch_size,
    shuffle=True,
    num_workers=opt.n_cpu,
)
# Test data loader
val_dataloader = DataLoader(
    ImageDataset("../../data/%s" % opt.dataset_name, transforms_=transforms_, unaligned=True, mode="test"),
    batch_size=5,
    shuffle=True,
    num_workers=1,
)

shuffle的参数设置错误导致，因为已经有batch_sample了，就不需要shuffle来进行随机的sample了，所以在这里的shuffle应该设置为FALSE才对。

应该改成：

# Training data loader
dataloader = DataLoader(
    ImageDataset("../../data/%s" % opt.dataset_name, transforms_=transforms_, unaligned=T

最低0.47元/天解锁文章