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 不匹配

LARC DL笔记(四):finetune resnet on food-101

target_blobs.size() == source_layer.blobs_size() (5 vs. 3) Incompatible number

查网上说是finetune的时候,source和目标不对应导致,但是换到server服务器的BVLC caffe就可以了,看来跟版本有一定关系

相关文章: