Reduce simplified

diff --git a/tools/caffe.cpp b/tools/caffe.cpp
index 431bf0e7d8e6b2be09eef099b488cc8b16c05e4e..6c0b5d49a635d1ed5adb03b09e914a4c5e5de98f 100644 (file)
--- a/tools/caffe.cpp
+++ b/tools/caffe.cpp
@@ -213,6 +213,7 @@ int train() {
   }
 
   if (gpus.size() > 1) {
+    Caffe::set_solver_count(gpus.size());
     caffe::P2PManager p2p_mgr(solver, gpus.size(), solver->param());
     p2p_mgr.Run(gpus);
   } else {