diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2015-01-07 20:41:09 -0800 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2015-02-05 14:49:21 -0800 |
| commit | 3b13846486875be2911716bcd1947dca71a97817 (patch) | |
| tree | 082b89cb75f1369635ca1365e41182a7f8730b40 /src/caffe/layers/argmax_layer.cpp | |
| parent | 648aed72acf1c506009ddb33d8cace40b75e176e (diff) | |
| download | caffeonacl-3b13846486875be2911716bcd1947dca71a97817.tar.gz caffeonacl-3b13846486875be2911716bcd1947dca71a97817.tar.bz2 caffeonacl-3b13846486875be2911716bcd1947dca71a97817.zip | |
Layer type is a string
Diffstat (limited to 'src/caffe/layers/argmax_layer.cpp')
| -rw-r--r-- | src/caffe/layers/argmax_layer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/argmax_layer.cpp b/src/caffe/layers/argmax_layer.cpp index 15e199eb..c4040cdc 100644 --- a/src/caffe/layers/argmax_layer.cpp +++ b/src/caffe/layers/argmax_layer.cpp @@ -58,6 +58,6 @@ void ArgMaxLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, } INSTANTIATE_CLASS(ArgMaxLayer); -REGISTER_LAYER_CLASS(ARGMAX, ArgMaxLayer); +REGISTER_LAYER_CLASS(ArgMax); } // namespace caffe |