output loss for caffenet and alexnet train/val models
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Sun, 10 Aug 2014 05:24:15 +0000 (22:24 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Sun, 10 Aug 2014 05:24:24 +0000 (22:24 -0700)
examples/imagenet/alexnet_train_val.prototxt
examples/imagenet/imagenet_train_val.prototxt

index f65f3e7..8d7977d 100644 (file)
@@ -336,4 +336,5 @@ layers {
   type: SOFTMAX_LOSS
   bottom: "fc8"
   bottom: "label"
+  top: "loss"
 }
index af28c14..ea1f907 100644 (file)
@@ -336,4 +336,5 @@ layers {
   type: SOFTMAX_LOSS
   bottom: "fc8"
   bottom: "label"
+  top: "loss"
 }