From b123fcc635317853cc9861f57c71d2f5e70ea8bc Mon Sep 17 00:00:00 2001
From: Evan Shelhamer <shelhamer@imaginarynumber.net>
Date: Sat, 9 Aug 2014 22:24:15 -0700
Subject: [PATCH] output loss for caffenet and alexnet train/val models

---
 examples/imagenet/alexnet_train_val.prototxt  | 1 +
 examples/imagenet/imagenet_train_val.prototxt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/examples/imagenet/alexnet_train_val.prototxt b/examples/imagenet/alexnet_train_val.prototxt
index f65f3e7..8d7977d 100644
--- 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
--- 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"
 }
-- 
2.7.4