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)

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

你可能感兴趣的:(机器学习)