该问题的产生是由于windows下多线程的问题,和DataLoader类有关
原因是windows 下 DataLoader加载数据集启用线程的数目这一块,是有bug的
trainloader = DataLoader(dataset_train,
batch_size=batch_size,
shuffle=True,
num_workers=1)
testloader = DataLoader(dataset_test,
batch_size=batch_size,
shuffle=False,
num_workers=1)
解决方法很简答,就是把原代码中的num_workers=1改为=0即可