Caffe 源碼閱讀(五) Solver.cpp

1、Solver类两个构造函数

Solver(const SolverParameter& param)

Solver(const string& param_file)

初始化两个类net_和test_net并调用Init()函数

2、Init函数

初始化网络

设置随机数种子 申请一块Net空间,test_net_指向这块空间

输入:SolverParameter 类型 param

3、训练网络 Solve函数

设置caffe mode

设置当前阶段(Train/Test)

调用PreSolve函数 Restore函数

调用一遍Test()判断内存是否够

对于每一次训练时的迭代:while(iter_++<param_.max_iter())

原文地址:https://www.cnblogs.com/Victory-walt/p/5591933.html