From: Yangqing Jia Date: Mon, 30 Sep 2013 00:29:05 +0000 (-0700) Subject: simple forward test X-Git-Url: https://git.ti.com/gitweb?p=jacinto-ai%2Fcaffe-jacinto.git;a=commitdiff_plain;h=8c618970155abcb3badd757dca152fd0922c8fab simple forward test --- diff --git a/src/caffe/test/test_simple_conv.cpp b/src/caffe/test/test_simple_conv.cpp index 3d849624..98dfa3c4 100644 --- a/src/caffe/test/test_simple_conv.cpp +++ b/src/caffe/test/test_simple_conv.cpp @@ -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 caffe_net(net_param, bottom_vec); + LOG(ERROR) << "Start Forward."; const vector*>& output = caffe_net.Forward(bottom_vec); LOG(ERROR) << "Forward Done."; EXPECT_EQ(output[0]->num(), 1);