num_works参数 就是进程的使用情况。
num_works 非零时,进入for循环会卡在这。
改为0,可以运行了。
把DataLoader中的参数num_workers 设为 0即可
参考资料
Debugger freezes stepping forward when using pytorch with workers (multiprocessing)
知乎 - pytorch dataloader 使用batch和 num_works参数的原理是什么? 原理解析
CSDN - PyTorch 训练时中遇到的卡住停住等问题 (推荐)
使用Pytorch dataloader时卡住 报错的解决方法
DataLoader, when num_worker >0, there is bug
解决pytorch DataLoader num_workers出现的问题