caffe常见问题

来源:互联网 发布:人文社科书籍推荐 知乎 编辑:程序博客网 时间:2024/06/05 13:27

1.显存不足
“error == cudaSuccess (2 vs. 0) out of memory”
减小该模型的train_val.prototxt中的batch_size(如果减到1还不行的话,准备换显卡吧)。

0 0