diff options
author | Yangqing Jia <jiayq84@gmail.com> | 2013-10-15 13:54:49 -0700 |
---|---|---|
committer | Yangqing Jia <jiayq84@gmail.com> | 2013-10-15 13:54:49 -0700 |
commit | b21ca1158c098230d99969b045b04696b1a565d5 (patch) | |
tree | 4acfd57b5cf466eb84ca9b6cea52d427b624b81e /src | |
parent | 7b49a585b8273b3d37c3ca3286612dfed697d34a (diff) | |
download | caffe-b21ca1158c098230d99969b045b04696b1a565d5.tar.gz caffe-b21ca1158c098230d99969b045b04696b1a565d5.tar.bz2 caffe-b21ca1158c098230d99969b045b04696b1a565d5.zip |
removed the net proto test that relies on external data
Diffstat (limited to 'src')
-rw-r--r-- | src/caffe/test/test_net_proto.cpp | 66 |
1 files changed, 0 insertions, 66 deletions
diff --git a/src/caffe/test/test_net_proto.cpp b/src/caffe/test/test_net_proto.cpp deleted file mode 100644 index 38b5b68d..00000000 --- a/src/caffe/test/test_net_proto.cpp +++ /dev/null @@ -1,66 +0,0 @@ -// Copyright 2013 Yangqing Jia - -#include <cuda_runtime.h> -#include <fcntl.h> -#include <google/protobuf/text_format.h> -#include <google/protobuf/io/zero_copy_stream_impl.h> -#include <gtest/gtest.h> - -#include <cstring> - -#include "caffe/blob.hpp" -#include "caffe/common.hpp" -#include "caffe/net.hpp" -#include "caffe/filler.hpp" -#include "caffe/proto/caffe.pb.h" -#include "caffe/util/io.hpp" - -#include "caffe/test/test_caffe_main.hpp" - -namespace caffe { - -template <typename Dtype> -class NetProtoTest : public ::testing::Test {}; - -typedef ::testing::Types<float, double> Dtypes; -TYPED_TEST_CASE(NetProtoTest, Dtypes); - -TYPED_TEST(NetProtoTest, TestSetup) { - NetParameter net_param; - ReadProtoFromTextFile("data/lenet.prototxt", &net_param); - // check if things are right - EXPECT_EQ(net_param.layers_size(), 10); - EXPECT_EQ(net_param.input_size(), 0); - - vector<Blob<TypeParam>*> bottom_vec; - - Net<TypeParam> caffe_net(net_param, bottom_vec); - EXPECT_EQ(caffe_net.layer_names().size(), 10); - EXPECT_EQ(caffe_net.blob_names().size(), 10); - - /* - // Print a few statistics to see if things are correct - for (int i = 0; i < caffe_net.blobs().size(); ++i) { - LOG(ERROR) << "Blob: " << caffe_net.blob_names()[i]; - LOG(ERROR) << "size: " << caffe_net.blobs()[i]->num() << ", " - << caffe_net.blobs()[i]->channels() << ", " - << caffe_net.blobs()[i]->height() << ", " - << caffe_net.blobs()[i]->width(); - } - */ - Caffe::set_mode(Caffe::CPU); - // Run the network without training. - LOG(ERROR) << "Performing Forward"; - caffe_net.Forward(bottom_vec); - LOG(ERROR) << "Performing Backward"; - LOG(ERROR) << caffe_net.Backward(); - - Caffe::set_mode(Caffe::GPU); - // Run the network without training. - LOG(ERROR) << "Performing Forward"; - caffe_net.Forward(bottom_vec); - LOG(ERROR) << "Performing Backward"; - LOG(ERROR) << caffe_net.Backward(); -} - -} // namespace caffe |