Caffe训练AlexNet网络模型——问题二

训练时,出现Check failed:error == cudaSuccess (2 vs. 0) out of memory,并且accruary = 0,如下图所示:

解决方法:将train_val.prototxt文件中的batch_size变小一点,如下图所示:

也可参见博客:

http://blog.csdn.net/u013066730/article/details/53784614

 

posted @ 2017-03-09 16:02  鹰杰  阅读(803)  评论(0编辑  收藏  举报