caffe源码解析 — train_net.cpp

train_net.cpp是caffe的主函数所在处，学习caffe时应该由此辐射到各个具体的函数去学习。

调用ReadProtoFromTextFileOrDie()从argv[1]中读入solver_param：

SolverParameter  solver_param

用solver_param去初始化solver，solverSGDSolver：

solver(solver_param)

调用solver的Solve函数，

solver.Solve(argv[2])

或

solver.Solve()

train_net.cpp

int main(int argc, char** argv) {
::google::InitGoogleLogging(argv[0]);
::google::SetLogDestination(0, "../tmp/");
if (argc < 2 || argc > 3) {
LOG(ERROR) << "Usage: train_net solver_proto_file [resume_point_file]";
return 1;
}
SolverParameter solver_param;
ReadProtoFromTextFileOrDie(argv[1], &solver_param);

LOG(INFO) << "Starting Optimization";
SGDSolver<float> solver(solver_param);
if (argc == 3) {
LOG(INFO) << "Resuming from " << argv[2];
solver.Solve(argv[2]);
} else {
solver.Solve();
}
LOG(INFO) << "Optimization Done.";

return 0;
}