summaryrefslogtreecommitdiff
path: root/docs
diff options
context:
space:
mode:
authorSergio <sguada@gmail.com>2014-09-06 16:30:32 -0700
committerSergio <sguada@gmail.com>2014-09-06 16:30:32 -0700
commit3578d91da9ea45e52c049c5bfaec83aedea3f06b (patch)
treec7306d068d39fa9f996098dea8470eb758617687 /docs
parent5eb8dd3558474536f115d4e951e08683cf6be4d9 (diff)
downloadcaffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.tar.gz
caffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.tar.bz2
caffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.zip
Added initial Hinge Loss
Diffstat (limited to 'docs')
-rw-r--r--docs/tutorial/layers.md32
1 files changed, 31 insertions, 1 deletions
diff --git a/docs/tutorial/layers.md b/docs/tutorial/layers.md
index 56fc338f..42ae5986 100644
--- a/docs/tutorial/layers.md
+++ b/docs/tutorial/layers.md
@@ -116,7 +116,37 @@ Loss drives learning by comparing an output to a target and assigning cost to mi
#### Hinge / Margin
-`HINGE_LOSS`
+* LayerType: `HINGE_LOSS`
+* CPU implementation: `./src/caffe/layers/hinge_loss_layer.cpp`
+* CUDA GPU implementation: `NOT_AVAILABLE`
+* Options (`HingeLossParameter hinge_loss_param`)
+ - Optional (default L1): `norm`, the norm used. Currently L1, L2
+* Inputs
+ - `n * c * h * w` Predictions
+ - `n * 1 * 1 * 1` Labels
+* Output
+ - `1 * 1 * 1 * 1` Computed Loss
+* Samples
+
+ # L1 Norm
+ layers {
+ name: "loss"
+ type: HINGE_LOSS
+ bottom: "pred"
+ bottom: "label"
+ }
+
+ # L2 Norm
+ layers {
+ name: "loss"
+ type: HINGE_LOSS
+ bottom: "pred"
+ bottom: "label"
+ top: "loss"
+ hinge_loss_param {
+ norm: L2
+ }
+ }
#### Sigmoid Cross-Entropy