set_mode to CPU in ArgMaxLayerTest constructor to avoid random Travis

failures.

diff --git a/src/caffe/test/test_argmax_layer.cpp b/src/caffe/test/test_argmax_layer.cpp
index 5286a0a..51d68c4 100644 (file)
--- a/src/caffe/test/test_argmax_layer.cpp
+++ b/src/caffe/test/test_argmax_layer.cpp
@@ -21,6 +21,7 @@ class ArgMaxLayerTest : public ::testing::Test {
       : 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;
@@ -59,7 +60,6 @@ TYPED_TEST(ArgMaxLayerTest, TestSetupMaxVal) {
 
 TYPED_TEST(ArgMaxLayerTest, TestCPU) {
   LayerParameter layer_param;
-  Caffe::set_mode(Caffe::CPU);
   ArgMaxLayer<TypeParam> layer(layer_param);
   layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
   layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_));
@@ -83,7 +83,6 @@ TYPED_TEST(ArgMaxLayerTest, TestCPU) {
 
 TYPED_TEST(ArgMaxLayerTest, TestCPUMaxVal) {
   LayerParameter layer_param;
-  Caffe::set_mode(Caffe::CPU);
   ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
   argmax_param->set_out_max_val(true);
   ArgMaxLayer<TypeParam> layer(layer_param);
@@ -110,7 +109,6 @@ TYPED_TEST(ArgMaxLayerTest, TestCPUMaxVal) {
 
 TYPED_TEST(ArgMaxLayerTest, TestCPUTopK) {
   LayerParameter layer_param;
-  Caffe::set_mode(Caffe::CPU);
   ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
   argmax_param->set_top_k(this->top_k_);
   ArgMaxLayer<TypeParam> layer(layer_param);
@@ -140,7 +138,6 @@ TYPED_TEST(ArgMaxLayerTest, TestCPUTopK) {
 
 TYPED_TEST(ArgMaxLayerTest, TestCPUMaxValTopK) {
   LayerParameter layer_param;
-  Caffe::set_mode(Caffe::CPU);
   ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
   argmax_param->set_out_max_val(true);
   argmax_param->set_top_k(this->top_k_);