DataLoader worker exited unexpectedly

Question

运行这段代码时，出现RuntimeError: DataLoader worker (pid(s) 2176) exited unexpectedly，请问怎么解决的啊，麻烦大家看一下

Answer 1

解决方法1：num_works = 0
就是在dataloader中设置，例如：train_loader = torch.utils.data.DataLoader(train_dataset,
batch_size=batch_size, shuffle=True,
num_workers=0)

解决方法2：在程序中加入
torch.multiprocessing.set_start_method("spawn")