summaryrefslogtreecommitdiff
path: root/tools
diff options
context:
space:
mode:
authorJun Shi <junshi@yahoo-inc.com>2016-01-22 09:58:37 -0800
committerJun Shi <junshi@yahoo-inc.com>2016-03-05 07:07:21 -0800
commit01528918c707df82e5910bea0270d7987db5abd8 (patch)
tree6a9b136eda30dd7b15a7d2daa6e874c22e2b4467 /tools
parent6a0b98768d4745714e31949b87382ff562be6724 (diff)
downloadcaffeonacl-01528918c707df82e5910bea0270d7987db5abd8.tar.gz
caffeonacl-01528918c707df82e5910bea0270d7987db5abd8.tar.bz2
caffeonacl-01528918c707df82e5910bea0270d7987db5abd8.zip
split p2psync::run()
Diffstat (limited to 'tools')
-rw-r--r--tools/caffe.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp
index 95b2f82c..5d9331f0 100644
--- a/tools/caffe.cpp
+++ b/tools/caffe.cpp
@@ -214,7 +214,7 @@ int train() {
if (gpus.size() > 1) {
caffe::P2PSync<float> sync(solver, NULL, solver->param());
- sync.run(gpus);
+ sync.Run(gpus);
} else {
LOG(INFO) << "Starting Optimization";
solver->Solve();