src/caffe/optimization/solver.hpp

   1 #ifndef CAFFE_OPTIMIZATION_SOLVER_HPP_
   2 #define CAFFE_OPTIMIZATION_SOLVER_HPP_
   3
   4 namespace caffe {
   5
   6 class Solver {
   7  public:
   8   explicit Solver(const SolverParameter& param)
   9       : param_(param) {}
  10   void Solve(Net* net);
  11
  12  protected:
  13   SolverParameter param_;
  14 };
  15
  16 }  // namspace caffe
  17
  18 #endif  // CAFFE_OPTIMIZATION_SOLVER_HPP_