当前位置: 代码迷 >> 综合 >> pytorch报错:ValueError: num_samples should be a positive integer value, but got num_samp=0
  详细解决方案

pytorch报错:ValueError: num_samples should be a positive integer value, but got num_samp=0

热度:5   发布时间:2023-12-14 05:47:03.0

跑baseline调试代码时遇到了一些问题,当时报了错,反复找不到原因,后来受到了下面这篇博客的启发终于解决了问题,我就将这个问题再总结一下:

https://blog.csdn.net/xnmc2014/article/details/85557384

出现的问题的地方可能是如下的地方:调用DataLoder时注意参数

self.train_dataloader = DataLoader(train_dataset, batch_size=TrainOption.train_batch_size, shuffle=TRUE,num_workers=TrainOption.data_load_worker_num)

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

 

  相关解决方案