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

来源:互联网 发布:javaweb邮件系统源码 编辑:程序博客网 时间:2024/06/13 07:57

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就可以了,看来跟版本有一定关系

阅读全文
1 0
原创粉丝点击