添加新的层后，使用经过训练的模型(微调)

Question

我想微调我的数据集上的bvlc_reference_caffenet模型。我成功地制作了模型。但现在，我试图增加2个卷积层，以检查它将如何执行。我尝试在"conv6_new"层之后添加名为"conv5"和"conv7_new"的层。我复制了"conv3"层，并将它重命名为"conv6_new"，将"conv5"重命名为"conv7_new"。我修改了相应的"ReLU"和"Pooling"层的名称。但我会跟着错误走。

有人能告诉我我在哪里犯错误吗？

F0620 17:29:49.967382 14194 insert_splits.cpp:29]未知的底部blob 'conv7_new‘(层’卷积7_new‘，底部指数0)

Answer 1

Layer "conv7_new"也不能有自己的输出("top")作为输入。确保您重命名，不仅是层的名称，而且是"top"和"bottom"的名称，这样做是有意义的：

如果你有"conv5" -> "conv6_new" -> "conv7_new" -> "fc"，

那么你应该有"conv6_new"的底部是"top" of "conv5"，"conv7_new"的底部是"top" of "conv6_new"，最后，"fc"的底部是您新的"conv7_new"层的"top"。

请注意，我忽略了可以通过设置与"ReLU"相同的"top"名称来计算“就地”的"bottom"。我也没有包括池层。