pytorch错误解决: BrokenPipeError: [Errno 32] Broken pipe

该问题的产生是由于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即可

你可能感兴趣的:(其他,pytorch,深度学习,人工智能)