machine-learning - 咖啡 |添加新层以使用训练模型后未知的底部 Blob (微调)

Question

我想微调我的数据集上的 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)

Answer 1

层“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” >。我也没有包括池化层。