pytorch运行报错:BrokenPipeError
在做机器学习训练时候出现的问题。错误原因:该问题的产生是由于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(t...
原创
2022-03-19 15:59:40 ·
1570 阅读 ·
0 评论