caffe - 在caffe中deploy.prototxt中需要定义dropout层吗？

Question

在caffe中的AlexNet实现中，我在deploy.prototxt文件中看到了以下层:

layer {
  name: "drop7"
  type: "Dropout"
  bottom: "fc7"
  top: "fc7"
  dropout_param {
    dropout_ratio: 0.5
  }
}

现在 dropout 的关键思想是在训练期间从神经网络中随机丢弃单元(连同它们的连接)。

这是否意味着我可以简单地从 deploy.prototxt 中删除这一层，因为该文件仅在测试期间使用？

Answer 1

是的。测试期间不需要辍学。

即使您包含一个 dropout 层，测试期间也不会发生任何特殊情况。见dropout forward pass的源码:

  if (this->phase_ == TRAIN) {
    // Code to do something
  } else {
    caffe_copy(bottom[0]->count(), bottom_data, top_data);  //Code to copy bottom blob to top blob
  }

如源代码所示，如​​果底部 blob 数据不在训练阶段，则将其复制到顶部 blob 数据内存。