您好，多线程dataloader 时，训练总是意外中断是什么原因?无法获取下一个batch数据。

[GYAN2019]

无报错，GPU 不在工作，CPU多进程在工作。请问如何解决。

[ge-xing]

截个图看看吧 我没遇到过这个问题，或者 train_process 设置小一点

[GYAN2019]

截个图看看吧 我没遇到过这个问题，或者 train_process 设置小一点

您好，感谢回复。基于您的工作想做一个回归问题的实验，输入双通道图像预测一个分布，调整了损失函数。其它没做大改动。

四个训练数据，一个验证数据。使用数据增强，就会出现下面的情况：显卡停了，进程在运行，batch数据怎么都加载不进来。试了nnUnet 单线程的dataloader, 数据增强关了，是可以的。检查了内存啥的都没啥问题。对多进程的操作一直不是很了解，是单纯的慢，还是哪里进程冲突了？

谢谢，祝好。

[GYAN2019]

only mirror的数据增强也是没问题的。是spatial transforms 单纯的慢吗？之前跑nnUnet 的时候好像没遇到这样的问题。