summaryrefslogtreecommitdiff
path: root/data
diff options
context:
space:
mode:
Diffstat (limited to 'data')
-rw-r--r--data/lenet.prototxt2
1 files changed, 1 insertions, 1 deletions
diff --git a/data/lenet.prototxt b/data/lenet.prototxt
index c8a99946..f5877ae4 100644
--- a/data/lenet.prototxt
+++ b/data/lenet.prototxt
@@ -114,7 +114,7 @@ layers {
}
layers {
layer {
- name: "prob"
+ name: "loss"
type: "softmax_loss"
}
bottom: "ip2"