Update solver.cpp

Added these two conditions before trying SetDevice param_.solver_mode() && param_.has_device_id()
HulkMaker · Feb 3, 2014 · 55cd18d · 55cd18d
1 parent 5962c59
commit 55cd18d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
@@ -40,7 +40,7 @@ Solver<Dtype>::Solver(const SolverParameter& param)
 template <typename Dtype>
 void Solver<Dtype>::Solve(const char* resume_file) {
   Caffe::set_mode(Caffe::Brew(param_.solver_mode()));
-  if (param_.solver_mode()) {
+  if (param_.solver_mode() && param_.has_device_id()) {
     Caffe::SetDevice(param_.device_id());
   }
   Caffe::set_phase(Caffe::TRAIN);