digits训练时报错 CUDNN_STATUS_SUCCESS (4 vs. 0)
2017-02-03 22:00
501 查看
解决:网络里的最后一个全连接层，没有设置num_output
坑爹的digits这一句我因为直接用的AlexNet,它注释写的，若把num_output注释掉的话系统会自己设置为你的图片类别数。。结果直接报错，还是得手动设置。。
之后又遇到这个错误，发现主要的原因还是batch_size设大了导致显存不够。。把batch_size调小就行了。
但是为啥用命令行训练的时候256x256的图片，我batch_size设为40都没问题，digits里面设20都出错，用的都是15才行？
坑爹的digits这一句我因为直接用的AlexNet,它注释写的，若把num_output注释掉的话系统会自己设置为你的图片类别数。。结果直接报错，还是得手动设置。。
之后又遇到这个错误，发现主要的原因还是batch_size设大了导致显存不够。。把batch_size调小就行了。
但是为啥用命令行训练的时候256x256的图片，我batch_size设为40都没问题，digits里面设20都出错，用的都是15才行？
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 
相关文章推荐
- 【caffe训练mobilenet】报错:Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ER
- 【caffe】Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERRO
- Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
- 【Caffe】Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
- caffe mnist测试Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNA
- caffe make runtest error（core dumped）Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)
- caffe make runtest error（core dumped）Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)
- caffe:Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
- Windows下编译CAFFE，status == CUDNN_STATUS_SUCCESS，关闭CUDNN
- Windows下编译CAFFE+CUDA, 运行时提示status == CUDNN_STATUS_SUCCESS错误
- 运行caffe时报错：Check failed: status == CUDNN_STATUS_SUCCESS
- 在ubuntu上配置cuda+cudnn+caffe(包括python和matlab接口)+digits
- faster-rcnn训练时出现error == cudaSuccess (30 vs. 0)
- caffe运行错误： im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- Keras+Theano后端，CPU、GPU、cuDNN加速对比（Dogs vs. Cats和mnist）
- Windows下编译CAFFE+CUDA, 运行时提示status == CUDNN_STATUS_SUCCESS错误
- could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR问题解决方案
- VS2013+CUDA7.5（+cudnn7.0）+Opencv3.0.0+caffe
- mxnet CUDNN_STATUS_ALLOC_FAILED 错误 cudnn出错
- tensorflow 1.4 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 错误的解决办法