summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 7a894bd)
raw | patch | inline | side by side (parent: 7a894bd)
author | Sergei Nikolaev <snikolaev@nvidia.com> | |
Sun, 14 Jan 2018 09:41:13 +0000 (01:41 -0800) | ||
committer | Sergei Nikolaev <snikolaev@nvidia.com> | |
Sun, 14 Jan 2018 09:41:13 +0000 (01:41 -0800) |
tools/caffe.cpp | patch | blob | history |
diff --git a/tools/caffe.cpp b/tools/caffe.cpp
index 431bf0e7d8e6b2be09eef099b488cc8b16c05e4e..6c0b5d49a635d1ed5adb03b09e914a4c5e5de98f 100644 (file)
--- a/tools/caffe.cpp
+++ b/tools/caffe.cpp
}
if (gpus.size() > 1) {
+ Caffe::set_solver_count(gpus.size());
caffe::P2PManager p2p_mgr(solver, gpus.size(), solver->param());
p2p_mgr.Run(gpus);
} else {