simple forward test
authorYangqing Jia <jiayq84@gmail.com>
Mon, 30 Sep 2013 00:29:05 +0000 (17:29 -0700)
committerYangqing Jia <jiayq84@gmail.com>
Mon, 30 Sep 2013 00:29:05 +0000 (17:29 -0700)
src/caffe/test/test_simple_conv.cpp

index 3d8496243c4a603b542645b1bd4c60d0a27bc1f1..98dfa3c4526a30983383caf6067bc0b0fc8cfbcc 100644 (file)
@@ -33,8 +33,10 @@ TYPED_TEST(NetProtoTest, TestLoadFromText) {
     EXPECT_LE(lena_image.cpu_data()[i], 1);
   }
 
+  Caffe::set_mode(Caffe::CPU);
   // Initialize the network, and then does smoothing
   Net<TypeParam> caffe_net(net_param, bottom_vec);
+  LOG(ERROR) << "Start Forward.";
   const vector<Blob<TypeParam>*>& output = caffe_net.Forward(bottom_vec);
   LOG(ERROR) << "Forward Done.";
   EXPECT_EQ(output[0]->num(), 1);