summaryrefslogtreecommitdiff
path: root/include/caffe/solver.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/caffe/solver.hpp')
-rw-r--r--include/caffe/solver.hpp14
1 files changed, 7 insertions, 7 deletions
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp
index 8d52785..2ecf539 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;
}