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