在做机器学习训练时候出现的问题。
错误原因:该问题的产生是由于windows下多线程的问题,和DataLoader类有关。将num_workers的数量修改为0即可。这是windows下的一个bug。
错误代码可以参考:
trainLoader=torch.utils.data.DataLoader(trainSet,batch_size=Bach_Size,shuffle=True,num_workers=2)
testLoader=torch.utils.data.DataLoader(testSet,batch_size=Bach_Size,shuffle=True,num_workers=2)
将num_workers的数量修改为0
trainLoader=torch.utils.data.DataLoader(trainSet,batch_size=Bach_Size,shuffle=True,num_workers=0)
testLoader=torch.utils.data.DataLoader(testSet,batch_size=Bach_Size,shuffle=True,num_workers=0)