diff options
author | Sergio <sguada@gmail.com> | 2014-09-06 16:30:32 -0700 |
---|---|---|
committer | Sergio <sguada@gmail.com> | 2014-09-06 16:30:32 -0700 |
commit | 3578d91da9ea45e52c049c5bfaec83aedea3f06b (patch) | |
tree | c7306d068d39fa9f996098dea8470eb758617687 /docs | |
parent | 5eb8dd3558474536f115d4e951e08683cf6be4d9 (diff) | |
download | caffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.tar.gz caffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.tar.bz2 caffeonacl-3578d91da9ea45e52c049c5bfaec83aedea3f06b.zip |
Added initial Hinge Loss
Diffstat (limited to 'docs')
-rw-r--r-- | docs/tutorial/layers.md | 32 |
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 |