内存溢出
0
收藏
在训练UNet的时候，疯狂溢出，训练三个batch就死掉（batch_size=2）
我用的环境是 CPU没有用GPU
而且试过别的网络（U2Net等等），也不知道为什么自己这里内存溢出现象很严重
（我这里就是在处理完图片以后，内存占用迅速上升到60%,我不理解的是为什么处理完以后不会降下来？）
希望高人能指点一下
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
其实应该慢慢往下降 但是简单粗暴的就是1
嗯，不想费事，就直接1
我觉得size改成1也行
如果是可以调小num_workers试试
是用的框架的DataLoader么？