当前位置: 代码迷 >> 综合 >> LARC DL笔记(四):finetune resnet on food-101
  详细解决方案

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

热度:13   发布时间:2024-01-11 15:22:28.0

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

  相关解决方案