Please enable JavaScript.
Coggle requires JavaScript to display documents.
Out of Memory (pytorch (0.利用nvidia-smi -lms检查是是否有显存泄露情况，即显存消耗越来越高, 2…
Out of Memory
-
-
-
-
:maple_leaf:尝试解决
:one:找到占用显存很大的代码,然后删除中间变量
:one:detect_face占用了5GB,而flip_test也占用了5GB
-
-
:three:使用train的时候输入的图片大小是[2, 3, 640, 640],而test的时候输入的图片大小是[1, 3,1080, 1920]
train中是直接提取出来就是640的.第一个是our/train/19/4440.jpg图片,我看看它是不是640的
-
:four:使用train的时候过一个layer显存都会自动下降,而使用test的时候不会
:two:换一台电脑
:one:使用自己电脑的GPU
:red_cross:因为我的电脑只有4GB显存,而且容易出现问题
:two:使用自己电脑的CPU
:red_cross:服务器都好多CPU都在跑,我只有4个CPU,显然不行
-
-
-