在跑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了。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐