赞
踩
今天学习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)
由于还在学习中具体的原因还不懂,也不能确定是不是这个参数的问题。只是尝试以下。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。