Question

我正在尝试使用caffe SqueezeNet。有2个版本（SqueezeNet_v1.0和SqueezeNet_v1.1）。我在SqueezeNet_v1.1 / train_val.prototxt（link）中注意到层“损失”和“准确性”阶段部分被评论：

layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "pool10"
  bottom: "label"
  top: "loss"
  #include {
  #  phase: TRAIN
  #}
}
layer {
  name: "accuracy"
  type: "Accuracy"
  bottom: "pool10"
  bottom: "label"
  top: "accuracy"
  #include {
  #  phase: TEST
  #}
}

有人可以解释一下原因吗？（不是另一个版本的情况，即SqueezeNet_v1.0 link）

Answer 1

通过评论阶段规范，您可以在培训和测试时间内获得准确度和损失值。

与SqueezeNet_v1.0版本不同，注释允许查看训练集的准确性（实际上是针对当前的训练批次）以及测试/验证集上的总损失，这可能对控制您的网络是否有用过度拟合。

caffe SqueezeNet：为什么阶段评论？

1 个答案: