diff options
| author | Sergio <sguada@gmail.com> | 2014-09-25 19:53:33 -0700 |
|---|---|---|
| committer | Sergio <sguada@gmail.com> | 2014-10-03 16:42:28 -0700 |
| commit | 8cd863e7cb5181cbbdfcf0dd75a78c80c56c1281 (patch) | |
| tree | d342596fd30a78b17800700e55dff6317ef7a56a /models | |
| parent | a9572b1bbe8dca87ec3e31804fee49301e066a94 (diff) | |
| download | caffeonacl-8cd863e7cb5181cbbdfcf0dd75a78c80c56c1281.tar.gz caffeonacl-8cd863e7cb5181cbbdfcf0dd75a78c80c56c1281.tar.bz2 caffeonacl-8cd863e7cb5181cbbdfcf0dd75a78c80c56c1281.zip | |
Fix OSX compilation for nvcc with opencv
Diffstat (limited to 'models')
| -rw-r--r-- | models/bvlc_reference_caffenet/train_val.prototxt | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/models/bvlc_reference_caffenet/train_val.prototxt b/models/bvlc_reference_caffenet/train_val.prototxt index 073d8aef..6cfeab10 100644 --- a/models/bvlc_reference_caffenet/train_val.prototxt +++ b/models/bvlc_reference_caffenet/train_val.prototxt @@ -330,11 +330,20 @@ layers { } } layers { - name: "accuracy" + name: "top-1" type: ACCURACY bottom: "fc8" bottom: "label" - top: "accuracy" + top: "top-1" + include: { phase: TEST } +} +layers { + name: "top-5" + type: ACCURACY + bottom: "fc8" + bottom: "label" + top: "top-5" + accuracy_param: { top_k: 5 } include: { phase: TEST } } layers { |