caffe训练过程中显示Check failed:error == cudaSuccess(2 vs. 0) out of memory

来源:互联网 发布:c语言逗号表达式的值 编辑:程序博客网 时间:2024/05/19 15:43

caffe训练过程中显示Check failed:error == cudaSuccess(2 vs. 0) out of memory,然后训练就结束。
解决:batch_size过大,把train和test的batch_size调小一点。
我的batch_size从4调整到2,顺利完成了训练过程。

阅读全文
0 0
原创粉丝点击