训练时,出现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