0x01 写在前面
有了googlenet的基础,resnet大体类似,但是出现了一些问题
0x02 问题记录
1、内存不足错误
resnet152,报failed to allocate 12845056 bytes on device 6
resnet占的内存太大,你batch size不能很多,不然16g也不够。。。
server上16/16都不行,于是调成8/16,可以跑
2、blobs.size 不匹配
target_blobs.size() == source_layer.blobs_size() (5 vs. 3) Incompatible number
查网上说是finetune的时候,source和目标不对应导致,但是换到server服务器的BVLC caffe就可以了,看来跟版本有一定关系