同一个train.py，没改任何代码，一会能跑，一会不能跑

问题描述

跑一个模型，加载预训练权重，只进行推理，没有修改任何代码，刚开始直接运行可以得到推理结果，然后再跑，程序又卡住了；然后我使用调试，一会能跑，一会跑到第五六个batch时，又卡住了。

解决方法

修改dataloader的num_workers。我程序原来workers设为的8，后面改为4就可以了。

原因

因为代码中使用了多线程（data loader的workers）。有时，多线程或多进程可能会导致死锁或资源竞争，从而使程序卡住。
尝试减少worker的数量或者设置为0，看看程序是否仍然卡住。