执行pytorch时报错:
RuntimeError: DataLoader worker (pid(s) 6432, 27660) exited unexpectedly
原因是:
在使用torch.utils.data.DataLoader时调用了多线程num_workers,
torch.utils.data.DataLoader(dataset, batch_size, shuffle=is_train, drop_last=is_train, num_workers=8)
需要将代码放置于 if __name__ == '__main__': 函数内执行。