diff options
Diffstat (limited to 'include/caffe')
-rw-r--r-- | include/caffe/solver.hpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp index 8d52785a..2ecf539b 100644 --- a/include/caffe/solver.hpp +++ b/include/caffe/solver.hpp @@ -283,19 +283,19 @@ Solver<Dtype>* GetSolver(const SolverParameter& param) { switch (type) { case SolverParameter_SolverType_SGD: - return new SGDSolver<Dtype>(param); + return new SGDSolver<Dtype>(param); case SolverParameter_SolverType_NESTEROV: - return new NesterovSolver<Dtype>(param); + return new NesterovSolver<Dtype>(param); case SolverParameter_SolverType_ADAGRAD: - return new AdaGradSolver<Dtype>(param); + return new AdaGradSolver<Dtype>(param); case SolverParameter_SolverType_RMSPROP: - return new RMSPropSolver<Dtype>(param); + return new RMSPropSolver<Dtype>(param); case SolverParameter_SolverType_ADADELTA: - return new AdaDeltaSolver<Dtype>(param); + return new AdaDeltaSolver<Dtype>(param); case SolverParameter_SolverType_ADAM: - return new AdamSolver<Dtype>(param); + return new AdamSolver<Dtype>(param); default: - LOG(FATAL) << "Unknown SolverType: " << type; + LOG(FATAL) << "Unknown SolverType: " << type; } return (Solver<Dtype>*) NULL; } |