blob: eef5290fe2eaf05ee09e0db1cf7a35f230ba0620 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#include <map>
#include <string>
#include "boost/scoped_ptr.hpp"
#include "google/protobuf/text_format.h"
#include "gtest/gtest.h"
#include "caffe/common.hpp"
#include "caffe/solver.hpp"
#include "caffe/solver_factory.hpp"
#include "caffe/test/test_caffe_main.hpp"
namespace caffe {
template <typename TypeParam>
class SolverFactoryTest : public MultiDeviceTest<TypeParam> {
protected:
SolverParameter simple_solver_param() {
const string solver_proto =
"train_net_param { "
" layer { "
" name: 'data' type: 'DummyData' top: 'data' "
" dummy_data_param { shape { dim: 1 } } "
" } "
"} ";
SolverParameter solver_param;
CHECK(google::protobuf::TextFormat::ParseFromString(
solver_proto, &solver_param));
return solver_param;
}
};
TYPED_TEST_CASE(SolverFactoryTest, TestDtypesAndDevices);
TYPED_TEST(SolverFactoryTest, TestCreateSolver) {
typedef typename TypeParam::Dtype Dtype;
typename SolverRegistry<Dtype>::CreatorRegistry& registry =
SolverRegistry<Dtype>::Registry();
shared_ptr<Solver<Dtype> > solver;
SolverParameter solver_param = this->simple_solver_param();
for (typename SolverRegistry<Dtype>::CreatorRegistry::iterator iter =
registry.begin(); iter != registry.end(); ++iter) {
solver_param.set_type(iter->first);
solver.reset(SolverRegistry<Dtype>::CreateSolver(solver_param));
EXPECT_EQ(iter->first, solver->type());
}
}
} // namespace caffe
|
