MCPcopy Create free account
hub / github.com/BVLC/caffe / InitSolverFromProtoString

Method InitSolverFromProtoString

src/caffe/test/test_solver.cpp:24–39  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

22
23 protected:
24 virtual void InitSolverFromProtoString(const string& proto) {
25 SolverParameter param;
26 CHECK(google::protobuf::TextFormat::ParseFromString(proto, &param));
27 // Set the solver_mode according to current Caffe::mode.
28 switch (Caffe::mode()) {
29 case Caffe::CPU:
30 param.set_solver_mode(SolverParameter_SolverMode_CPU);
31 break;
32 case Caffe::GPU:
33 param.set_solver_mode(SolverParameter_SolverMode_GPU);
34 break;
35 default:
36 LOG(FATAL) << "Unknown Caffe mode: " << Caffe::mode();
37 }
38 solver_.reset(new SGDSolver<Dtype>(param));
39 }
40
41 shared_ptr<Solver<Dtype> > solver_;
42};

Callers 1

TYPED_TESTFunction · 0.45

Calls 1

resetMethod · 0.45

Tested by

no test coverage detected