ValueError: num_samples should be a positive integer value, but got num_samples=0

        我在用 torch.utils.data.DataLoader加载数据时,报了一个这样的错:

        排查了dataset的代码,发现输出是正常的,最后想起来在修改代码时,有一处地方没有改:

        这里我在修改代码时,这个参数没有被用到了,所以这里输出的 len 为 0 。DataLoader读出来后,就报了 got num_samples=0,将这个修改后就正常运行了。

你可能感兴趣的:(python,pytorch,开发语言)