- 博客(6)
- 收藏
- 关注
原创 python模型训练在dataloader处卡死
1、前一进程还未处理完,又进入下一个导致互锁:在每个Epoch后,或每2、个 batch 后暂停一下: time.sleep(0.003);4、多进程导致互锁问题:减少进程数或者不用,num_workers=0/1。3、内存问题:使用开关,pin_memory=True/False。5、OpenCV与Pytorch互锁;6、batch size 的设置;7、查看是否存在循环;
2023-10-25 21:23:20
576
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人