caffe 训练问题集锦

1、Check failed: target_blobs.size() == source_layer.blobs_size() (2 vs. 1) Incompatible number of blobs for layer conv0

测试的时候出现此问题

2、cudaSuccess(2 vs. 0) out of memory,如下图

这种问题为显存不足导致的,出现这种问题有以下几种可能

1)若出现在train阶段,则train的batch_size设置的太大,此时改小batch_size一般就能解决

2)若出现在train过程中的test阶段,则test的batch_size设置的太大,此时改小batch_size一般就能解决

若改小test的batch_size还不能解决此问题是因为在整个train过程中test的时候train所占用的batch_size的空间并没有释放,所以train中的test的batch_size的大小为train的batch_size加上test的batch_size,这种情况就只能先train,train结束之后再test

3)若改小batch_size还是不能解决此情况,那么有可能是输入图片太大导致的,这时候在生成lmdb的时候把图片尺寸设置的小一点就可以了

4)若以上三种均不能解决,那么有可能是网络太复杂导致的,这时可以更换简单一点的网络去训练,或者换显存更大一点的GPU

3、

猜你喜欢

转载自blog.csdn.net/zhanghenan123/article/details/81115385