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

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

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

View File

@ -53,6 +53,11 @@ class AllreduceMock : public AllreduceRobust {
this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "CheckPoint");
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:
// key to identify the mock stage