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

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

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

images, labels = dataiter.next()

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

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

报错解决。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐