diff options
author | Jun Shi <junshi@yahoo-inc.com> | 2016-01-22 09:58:37 -0800 |
---|---|---|
committer | Jun Shi <junshi@yahoo-inc.com> | 2016-03-05 07:07:21 -0800 |
commit | 01528918c707df82e5910bea0270d7987db5abd8 (patch) | |
tree | 6a9b136eda30dd7b15a7d2daa6e874c22e2b4467 /include | |
parent | 6a0b98768d4745714e31949b87382ff562be6724 (diff) | |
download | caffeonacl-01528918c707df82e5910bea0270d7987db5abd8.tar.gz caffeonacl-01528918c707df82e5910bea0270d7987db5abd8.tar.bz2 caffeonacl-01528918c707df82e5910bea0270d7987db5abd8.zip |
split p2psync::run()
Diffstat (limited to 'include')
-rw-r--r-- | include/caffe/parallel.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/caffe/parallel.hpp b/include/caffe/parallel.hpp index 85fc2b55..6c496c88 100644 --- a/include/caffe/parallel.hpp +++ b/include/caffe/parallel.hpp @@ -93,7 +93,10 @@ class P2PSync : public GPUParams<Dtype>, public Solver<Dtype>::Callback, return solver_; } - void run(const vector<int>& gpus); + void Run(const vector<int>& gpus); + void Prepare(const vector<int>& gpus, + vector<shared_ptr<P2PSync<Dtype> > >* syncs); + inline const int initial_iter() const { return initial_iter_; } protected: void on_start(); |