projects / platform / upstream / caffeonacl.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 7dc6463)
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 patch | blob | history
examples/imagenet/imagenet_train_val.prototxt patch | blob | history

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"
 }
Domain: Machine Learning / ML Framework;