Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ok

Browse Source
This commit is contained in:
tqchen 2015-01-16 10:12:47 -08:00
parent aebb7998a3
commit f5245c615c
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rabit-learn/kmeans/.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
kmeans
*.mpi

5
src/allreduce_mock.h
View File

@ -53,6 +53,11 @@ class AllreduceMock : public AllreduceRobust {
this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "CheckPoint"); this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "CheckPoint");
AllreduceRobust::CheckPoint(global_model, local_model); AllreduceRobust::CheckPoint(global_model, local_model);
} }
virtual void LazyCheckPoint(const ISerializable *global_model) {
this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "LazyCheckPoint");
AllreduceRobust::LazyCheckPoint(global_model);
}
private: private:
// key to identify the mock stage // key to identify the mock stage