zoukankan      html  css  js  c++  java
  • caffe进行finetune时出现"shapeequals(proto) shape mismatch (reshape not set)"的解决办法

    声明加载的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的目的刚好也在于此么。

  • 相关阅读:
    hdu 4577 X-Boxes 大数
    hdu 4576 Robot 概率DP
    将IP地址转化为整数
    MyISAM压缩表
    yii2 模态框
    MySQL数据库设计
    foreach循环赋值问题
    实用的网站
    判断链接地址是否有效
    tp5获取配置文件信息
  • 原文地址:https://www.cnblogs.com/yyxf1413/p/7919310.html
Copyright © 2011-2022 走看看