我在用 torch.utils.data.DataLoader加载数据时,报了一个这样的错:
排查了dataset的代码,发现输出是正常的,最后想起来在修改代码时,有一处地方没有改:
这里我在修改代码时,这个参数没有被用到了,所以这里输出的 len 为 0 。DataLoader读出来后,就报了 got num_samples=0,将这个修改后就正常运行了。
我在用 torch.utils.data.DataLoader加载数据时,报了一个这样的错:
排查了dataset的代码,发现输出是正常的,最后想起来在修改代码时,有一处地方没有改:
这里我在修改代码时,这个参数没有被用到了,所以这里输出的 len 为 0 。DataLoader读出来后,就报了 got num_samples=0,将这个修改后就正常运行了。