内存溢出
方草木明 发布于2021-05 浏览:804 回复:5
0
收藏
快速回复

在训练UNet的时候,疯狂溢出,训练三个batch就死掉(batch_size=2)

我用的环境是 CPU没有用GPU

而且试过别的网络(U2Net等等),也不知道为什么自己这里内存溢出现象很严重

(我这里就是在处理完图片以后,内存占用迅速上升到60%,我不理解的是为什么处理完以后不会降下来?)

 

希望高人能指点一下

收藏
点赞
0
个赞
共5条回复 最后由七年期限回复于2021-05
#6七年期限回复于2021-05
#5 189******30回复
嗯,不想费事,就直接1

其实应该慢慢往下降  但是简单粗暴的就是1

0
#5189******30回复于2021-05
#4 七年期限回复
我觉得size改成1也行

嗯,不想费事,就直接1

0
#4七年期限回复于2021-05
#3 189******30回复
如果是可以调小num_workers试试

我觉得size改成1也行

0
#3189******30回复于2021-05

如果是可以调小num_workers试试

0
#2189******30回复于2021-05

是用的框架的DataLoader么?

0
TOP
切换版块