LOG(INFO) << "Initializing solver from parameters: " << std::endl
<< param.DebugString();
param_ = param;
+ if (param_.solver_mode() == SolverParameter_SolverMode_GPU &&
+ param_.has_device_id()) {
+ Caffe::SetDevice(param_.device_id());
+ }
+ Caffe::set_mode(Caffe::Brew(param_.solver_mode()));
if (param_.random_seed() >= 0) {
Caffe::set_random_seed(param_.random_seed());
}
LOG(INFO) << "Solver scaffolding done.";
}
-
template <typename Dtype>
void Solver<Dtype>::Solve(const char* resume_file) {
- Caffe::set_mode(Caffe::Brew(param_.solver_mode()));
- if (param_.solver_mode() == SolverParameter_SolverMode_GPU &&
- param_.has_device_id()) {
- Caffe::SetDevice(param_.device_id());
- }
Caffe::set_phase(Caffe::TRAIN);
LOG(INFO) << "Solving " << net_->name();
PreSolve();