output loss for caffenet and alexnet train/val models

diff --git a/examples/imagenet/alexnet_train_val.prototxt b/examples/imagenet/alexnet_train_val.prototxt
index f65f3e7..8d7977d 100644 (file)
--- a/examples/imagenet/alexnet_train_val.prototxt
+++ b/examples/imagenet/alexnet_train_val.prototxt
@@ -336,4 +336,5 @@ layers {
   type: SOFTMAX_LOSS
   bottom: "fc8"
   bottom: "label"
+  top: "loss"
 }

diff --git a/examples/imagenet/imagenet_train_val.prototxt b/examples/imagenet/imagenet_train_val.prototxt
index af28c14..ea1f907 100644 (file)
--- a/examples/imagenet/imagenet_train_val.prototxt
+++ b/examples/imagenet/imagenet_train_val.prototxt
@@ -336,4 +336,5 @@ layers {
   type: SOFTMAX_LOSS
   bottom: "fc8"
   bottom: "label"
+  top: "loss"
 }