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

最新推荐文章于 2024-09-24 15:42:21 发布

TracyGC

最新推荐文章于 2024-09-24 15:42:21 发布

阅读量676

点赞数 8

分类专栏：研究与思考文章标签：深度学习

本文链接：https://blog.csdn.net/tracygc/article/details/136286738

版权

研究与思考专栏收录该内容

37 篇文章 2 订阅

订阅专栏

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

摘要由CSDN通过智能技术生成

在跑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=True),
    batch_size=opt.batch_size,
    shuffle=False,
    num_workers=opt.n_cpu,
)
# Training data loader
dataloader = DataLoader(
    ImageDataset("../../data/%s" % opt.dataset_name, transforms_=transforms_, unaligned=True),
    batch_size=opt.batch_size,
    shuffle=False,
    num_workers=opt.n_cpu,
)