raise StopIteration .. StopIteration---next(iter(data.DataLoader())报错StopIteration

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
在这里插入图片描述

以上就是全部内容

你可能感兴趣的:(深度学习,python,深度学习,神经网络)