pytorch 的 多线程dataloader ，bug： DataLoader worker (pid(s) {}) exited unexpectedly

看了网上很多的教程 ，都是 将线程数改为0，  一直不理解为什么同样的环境， 我另外一个代码 可以跑多线程，  后来发现是main 函数的问题 ，   

不要把 训练部分的代码 全部放在 main函数if __name__ == "__main__":    里面！！！！！！！， 另外写一个函数 比如  def  train() :   将 训练代码放在这里面， 再用main函数调用：

如下图：  

、

我将所有的训练代码放在 train()里面 ，这样就可以使用多线程了， 我也不知道什么原因 。 我的情况是这样的。