今天用Pytorch训练一个小代码,加载数据的地方
self.train_loader = torch.utils.data.DataLoader(self.train_gaze, batch_size, shuffle=True, num_workers=workers)
一开始设置workers为4,报出如下错误
RuntimeError: DataLoader worker (pid(s) ****, ****, ****, ****) exited unexpectedly
后来把workers改成2还是出错,直到最后把workers改成0才不出错。电脑不行啊!
当报错时应该考虑一下是不是设置的太大或者不支持多线程进行操作。