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)

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


版权声明:本文为u011573853原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。