用GPU做caffe训练提示 out of memory

来源:互联网 发布:听考研英语单词软件 编辑:程序博客网 时间:2024/06/16 23:10

用GPU做caffe训练提示 out of memory

深度学习中训练时经常会出现 out of memory,出现这种情况的解决方法三种:

一、调小batch_size的值;

二、输入图片变小一些,可以使用resize;

三、换显卡。

原创粉丝点击