Pytorch中TypeError: 'DataLoader' object is not subscriptable错误

今天学习pytorch遇到以下问题

TypeError: 'DataLoader' object is not subscriptable

一开始设置的参数如下

cifar_train = DataLoader(cifar_train,batch_size=batchSize,shuffle=True)

就会报上面的错误,后来加上num_workers,莫名的好了

cifar_train = DataLoader(cifar_train,batch_size=batchSize,shuffle=True,num_workers=0)

由于还在学习中具体的原因还不懂,也不能确定是不是这个参数的问题。只是尝试以下。

发布了125 篇原创文章 · 获赞 63 · 访问量 22万+

猜你喜欢

转载自blog.csdn.net/u011573853/article/details/105580236