dataloader 的bug

Posted 2020-12-19 rabitvision

今天，为了对代码进行测试，使用了较小的数据。实际数据的大小是2， 也就是如下函数返回值是2.

def __len__(self):
    return self.length

使用dataloader时，batchsize=4。

运行 next(loader)， 就会出现卡死的现象。原来是batchsize 不能大于数据长度。