
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
shuffle=True, num_workers=2)
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
download=True, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
shuffle=False, num_workers=2)
经查资料
把num_workers=2改为 num_workers=0即可正常运行
本文介绍了一个关于PyTorch中CIFAR10数据集加载的问题及解决方案,通过调整num_workers参数,解决了数据加载过程中遇到的运行错误。
2409





