您的位置:首页 > Web前端

caffe进行finetune时出现"shapeequals(proto) shape mismatch (reshape not set)"的解决办法

2017-11-29 11:17 423 查看
声明:加载的caffemodel会根据你的net.prototxt文件里的各个layer的name来进行参数赋值。

错误:[Caffe]: Check failed: ShapeEquals(proto) shape mismatch (reshape not set)

原因:从caffemodel加载进来的参数和你的输入或者输出不匹配。

示例:比如对于全连接层fc6,caffemodel中的bottom是1*256*6*6,而在你修改后的网络结构中,bottom是1*512*6*6,此时就会出现上述的错误。由于输入的shape是mismatch的。

解决:修改网络结构的同时,修改相应的layer的name,这样子在加载caffemodel的时候就不会加载pre-train的参数进来,从而避免参数和数据不匹配,但是这样做也不可避免的重新初始化了部分参数,不过fine-tune的目的刚好也在于此么。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐