代码之家  ›  专栏  ›  技术社区  ›  Nima

caffe挤压网:为什么phase发表评论?

  •  0
  • Nima  · 技术社区  · 6 年前

    我正在尝试使用咖啡挤压网。有两个版本(SqueezeNet\u v1.0和SqueezeNet\u v1.1)。我注意到在SqueezeNet\u v1.1/train\u 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\u v1.0,情况并非如此 link )

    1 回复  |  直到 6 年前
        1
  •  2
  •   Lemm Ras    6 年前

    通过注释阶段规范,您可以在培训和测试期间获得精度和损耗值。

    与SqueezeNet\u v1.0版本不同,注释允许查看训练集(实际上是当前训练批次)的准确性和测试/验证集的总损失,这可能有助于控制网络是否过度匹配。