您的位置:首页 > Web前端

caffe在训练时遇到:Check failed: error == cudaSuccess (2 vs. 0) out of memory

2017-02-23 23:32 549 查看
batch_size太大了,一次性读入的图片太多了,所以就超出了显存。因此需要将train.prototxt中的文件train和test的batch_size调小一点。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  caffe
相关文章推荐