AttributeError: ‘_MultiProcessingDataLoaderIter‘ object has no attribute ‘next‘

用vision transformer跑Cifar-10分类代码的时候，报了个错：

**AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute 'next'**

检查过程中发现报错的代码在这里：

images, labels = dataiter.next()

原因：pytorch版本问题。
修改方案如下：

将
images, labels = dataiter.next()
修改为
images, labels = next(dataiter)

报错解决。