在跑pytorch代码时候根据github提供的源码克隆到自己电脑上跑,出现各种bug,经过替换几个python版本之后还是发现错误,依然报错:
ValueError: num_samples should be a positive integer value, but got num_samples=0
网上搜索发现有小伙伴出现类似情况(参考以下博客解决):
https://blog.csdn.net/qq_38247544/article/details/106651404
将
train_loader = DataLoader(train, batch_size=batch_size, shuffle=True, num_workers=8, pin_memory=True)
#此处shuffle=True改为False就不报错了
因为已经有batch_sample了,就不需要shuffle来进行随机的sample了。