_queue.Empty The above exception was the direct cause of the following exception:
RuntimeError: DataLoader worker (pid(s) 18392) exited unexpectedly
原因:还是
DataLoader的num_workers问题
DataLoader(salobj_dataset, batch_size=batch_size, shuffle=True, num_workers=0)
num_workers为1时,会单独建一个数据加载的进程,只有
num_workers改为0,将在主进程中加载数据,可以帮助确定是否为多进程问题
最后解决:
把num_workers改为0报错没有了。