#未读取到样本

完整报错
Traceback (most recent call last):
File “D:/LanduseDataset/pytorch_segmentation/segmentation_models/train_upernet.py”, line 292, in
main()
File “D:/LanduseDataset/pytorch_segmentation/segmentation_models/train_upernet.py”, line 73, in main
train_loader = DataLoader(train_dataset, batch_size=8, shuffle=True, num_workers=8,pin_memory=True)
File “C:\Users\laoji.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py”, line 277, in init
sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type]
File “C:\Users\laoji.conda\envs\pytorch\lib\site-packages\torch\utils\data\sampler.py”, line 97, in init
raise ValueError("num_samples should be a positive integer "
ValueError: num_samples should be a positive integer value, but got num_samples=0
Process finished with exit code 1

解决方案

检查数据集路径

Logo

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

更多推荐