Skip to content

Commit

Permalink
solver.hpp: add \briefs
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffdonahue authored and shelhamer committed Sep 3, 2014
1 parent 19cf385 commit 4572663
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions include/caffe/solver.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,12 @@

namespace caffe {

/**
* @brief An interface for classes that perform optimization on Net%s.
*
* Requires implementation of ComputeUpdateValue to compute a parameter update
* given the current state of the Net parameters.
*/
template <typename Dtype>
class Solver {
public:
Expand Down Expand Up @@ -57,6 +63,10 @@ class Solver {
};


/**
* @brief Optimizes the parameters of a Net using
* stochastic gradient descent (SGD) with momentum.
*/
template <typename Dtype>
class SGDSolver : public Solver<Dtype> {
public:
Expand Down

0 comments on commit 4572663

Please sign in to comment.