projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
f48cead
)
Make class ArgMaxLayerTest derive from CPUDeviceTest
author
Felix Abecassis
<fabecassis@nvidia.com>
Tue, 26 May 2015 18:21:46 +0000
(11:21 -0700)
committer
Felix Abecassis
<fabecassis@nvidia.com>
Tue, 26 May 2015 20:55:49 +0000
(13:55 -0700)
src/caffe/test/test_argmax_layer.cpp
patch
|
blob
|
history
diff --git
a/src/caffe/test/test_argmax_layer.cpp
b/src/caffe/test/test_argmax_layer.cpp
index
3487d42
..
895c3d3
100644
(file)
--- a/
src/caffe/test/test_argmax_layer.cpp
+++ b/
src/caffe/test/test_argmax_layer.cpp
@@
-13,13
+13,12
@@
namespace caffe {
template <typename Dtype>
-class ArgMaxLayerTest : public
::testing::Test
{
+class ArgMaxLayerTest : public
CPUDeviceTest<Dtype>
{
protected:
ArgMaxLayerTest()
: blob_bottom_(new Blob<Dtype>(10, 20, 1, 1)),
blob_top_(new Blob<Dtype>()),
top_k_(5) {
- Caffe::set_mode(Caffe::CPU);
Caffe::set_random_seed(1701);
// fill the values
FillerParameter filler_param;