Merge pull request BVLC#5393 from jfolz/master

Multi GPU training from Python can use any solver
W-Jilly · Mar 13, 2017 · 317d162 · 317d162
2 parents e687a71 + 1d3e6e4
commit 317d162
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
@@ -288,7 +288,7 @@ void Solver_add_callback(Solver<Dtype> * solver, bp::object on_start,
 }
 
 // Seems boost cannot call the base method directly
-void Solver_add_nccl(SGDSolver<Dtype>* solver
+void Solver_add_nccl(Solver<Dtype>* solver
 #ifdef USE_NCCL
   , NCCL<Dtype>* nccl
 #endif