raise StopIteration … StopIteration—next(iter(data.DataLoader())报错StopIteration

raise StopIteration … StopIteration—next(iter(data.DataLoader())报错StopIteration

原因

在使用next(iter(data.DataLoader())时报错raise StopIteration StopIteration，这是因为next()在访问一个已经迭代完的迭代器。这说明dataloader中的数据在一个epoch后，里面的数据都被取完了，进而导致空迭代器报错。

解决方法

保证每个epoch都使用迭代器封装一次dataloader

以上就是全部内容