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;
}
CHECK(Caffe::root_solver());
string model_filename;
switch (param_.snapshot_format()) {
- case caffe::SolverParameter_SnapshotFormat_BINARYPROTO:
- model_filename = SnapshotToBinaryProto();
- break;
- case caffe::SolverParameter_SnapshotFormat_HDF5:
- model_filename = SnapshotToHDF5();
- break;
- default:
- LOG(FATAL) << "Unsupported snapshot format.";
+ case caffe::SolverParameter_SnapshotFormat_BINARYPROTO:
+ model_filename = SnapshotToBinaryProto();
+ break;
+ case caffe::SolverParameter_SnapshotFormat_HDF5:
+ model_filename = SnapshotToHDF5();
+ break;
+ default:
+ LOG(FATAL) << "Unsupported snapshot format.";
}
SnapshotSolverState(model_filename);