Caffe模型调试中的错误总结

错误:Cannot copy param 0 weights from layer '...'; shape mismatch

即model和prototxt定义的模型不一样。通常是fine-tune时图片尺寸有了调整,或者改了网络结构想要train时去掉weight参数即可。

 

错误:Check failed: error == cudaSuccess (2 vs. 0) out of memory

错误提示GPU memery不足。首先可以用nvidia-smi来检查有没有多余程序占用了空间

如果没有可关闭的程序的话,可以在prototxt中把batch_size调小一点。

 

错误:eltwise_layer.cpp:34   Check failed: bottom[i]->shape() == bottom[0]->shape() 

这个错误就是在执行element-wise运算的时候(比如concat),操作数维度不一致。往上追溯错误日志,可以看到相加的两个元素,根据需要对其维度进行调整。

 

错误:math_functions.cu:79 Check failed: error == cudaSuccess (77 vs. 0) an illegal memory access was encountered

一开始以为还是batch_size调小点就行,结果调到1都不行。网上查找了一些资料,发现出现这个错误的原因非常多样,共同的结果就是内存访问出现问题。经过较长时间的debug,将某一层的depthconv换成conv后不再报错。

 

你可能感兴趣的:(深度学习)