训练时,出现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
hust-yingjie 2017-03-09 16:02 原文
训练时,出现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