【问题标题】:Cannot copy param 0 weights from layer 'conv1/7x7_s2/bn'; shape mismatch. Caffe Error无法从图层“conv1/7x7_s2/bn”复制参数 0 权重;形状不匹配。咖啡错误
【发布时间】:2018-11-03 11:52:57
【问题描述】:

我在 Digits 上训练了预训练的 GoogleNet(caffe)。培训进展顺利,Digits UI 中的测试显示没有任何问题。但是当我导出模型并尝试使用 example.py 提供的数字时。出现了这个错误。

无法从层“conv1/7x7_s2/bn”复制参数 0 权重;形状错配 H。源参数形状为 1 64 1 1 (64);目标参数形状为 64 (64)。从 sc 学习这一层的参数 ratch 而不是从保存的网络中复制,重命名图层。

我已经检查了 deploy.prototxt 和 train_val.prototxt。尺寸正确。

有什么建议吗?

【问题讨论】:

    标签: machine-learning neural-network deep-learning caffe nvidia-digits


    【解决方案1】:

    尝试为图层'conv1/7x7_s2/bn' 编辑您的'deploy.prototxt'。添加

    param {
      share_mode: PERMISSIVE
    }
    

    对于层的每个可训练参数。

    更多信息请参见caffe.proto

    【讨论】:

    • deploy.prototxt 中的每一层?为什么 example.py of digits 无法对其进行分类。我的意思是我使用数字来训练但不能使用 example.py 来测试它。
    • @asiaandudeCom 您只需将其添加到问题层的param。我不知道您为什么会收到此错误-关于维度的问题...
    猜你喜欢
    • 1970-01-01
    • 2017-10-15
    • 2018-04-20
    • 1970-01-01
    • 2020-07-02
    • 1970-01-01
    • 2015-09-24
    • 1970-01-01
    • 2015-11-06
    相关资源
    最近更新 更多