我想微调我的数据集上的 bvlc_reference_caffenet
模型。
我成功地对模型进行了微调。但现在我尝试添加 2 个卷积层来检查它的执行情况。我尝试在 "conv5"
层之后添加名称为 "conv6_new"
和 "conv7_new"
的层。我复制了 "conv3"
图层并通过将其重命名为 "conv6_new"
并将 "conv5"
重命名为 "conv7_new"
进行粘贴>。我修改了相应的“ReLU”和“Pooling”层的名称。
但我收到以下错误。
谁能告诉我我在哪里犯了错误?
F0620 17:29:49.967382 14194 insert_splits.cpp:29] Unknown bottom blob 'conv7_new' (layer 'conv7_new', bottom index 0)
最佳答案
层“conv7_new”
也不能将其自己的输出(“top”)作为输入。确保您不仅重命名图层的名称,还以有意义的方式重命名“top”
和“bottom”
:
如果您有 "conv5"
-> "conv6_new"
-> "conv7_new"
-> "fc"
,
那么“conv6_new”
的底部应该是“conv5”
的“top”
,而“conv7_new”
code> 的底部是 “conv6_new”
的 “top”
。最后,“fc”
的底部是新“conv7_new”
层的“top”
。
请注意,我忽略了可以通过将其 "top"
名称设置为与 "bottom"
相同来“就地”计算的“ReLU” >。我也没有包括池化层。
关于machine-learning - 咖啡 |添加新层以使用训练模型后未知的底部 Blob (微调),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37922730/