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:
4748845
)
simple forward test
author
Yangqing Jia
<jiayq84@gmail.com>
Mon, 30 Sep 2013 00:29:05 +0000
(17:29 -0700)
committer
Yangqing Jia
<jiayq84@gmail.com>
Mon, 30 Sep 2013 00:29:05 +0000
(17:29 -0700)
src/caffe/test/test_simple_conv.cpp
patch
|
blob
|
history
diff --git
a/src/caffe/test/test_simple_conv.cpp
b/src/caffe/test/test_simple_conv.cpp
index
3d84962
..
98dfa3c
100644
(file)
--- 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<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);