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
·
⭕今天在调试代码的时候一直出现这个报错,反复找不到原因
❗后面上网查询许多资料,询问大佬终于解决!!
❗出现的问题的地方是如下的地方:调用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才对。
希望对您有所帮助!!!!
更多推荐




所有评论(0)