Pytorch 报错:RuntimeError: DataLoader worker (pid(s) 2420, 11236, 7252, 3104) exited unexpectedly
并没有显示出是哪一行代码出错,但是根据提示中的DataLoader worker可知是多线程的问题,对应到代码中应该就是下图这里:
num_workers = 4
train_iter = torch.utils.data.DataLoader(mnist_train, batch_size=batch_size, shuffle=True, num_workers=num_workers)
test_iter