在调试的过程中,已经把ssd_train_resnet18_kitti.txt文件中的batch_size_per_gpu参数调至2甚至1,轮数10轮,但是每次都是1轮都没跑完就显示ran out of memory(如上图),我们用nvidia-smi查看gpu显存,显示的是还有还有余量(如上图)。但每次都是一训练一轮都没跑完就报错停下来。请问老师这是什么原因,并如何解决呢
在调试的过程中,已经把ssd_train_resnet18_kitti.txt文件中的batch_size_per_gpu参数调至2甚至1,轮数10轮,但是每次都是1轮都没跑完就显示ran out of memory(如上图),我们用nvidia-smi查看gpu显存,显示的是还有还有余量(如上图)。但每次都是一训练一轮都没跑完就报错停下来。请问老师这是什么原因,并如何解决呢