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

fix federated learning plugin (#8027)

Browse Source
This commit is contained in:
Rong Ou 2022-06-23 17:41:07 -07:00 committed by GitHub
parent a68580e2a7
commit 0725fd6081
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
plugin/federated/engine_federated.cc
View File

@ -84,17 +84,14 @@ class FederatedEngine : public IEngine {
}
}
int LoadCheckPoint(Serializable *global_model, Serializable *local_model = nullptr) override {
int LoadCheckPoint() override {
return 0;
}
void CheckPoint(const Serializable *global_model,
const Serializable *local_model = nullptr) override {
void CheckPoint() override {
version_number_ += 1;
}
void LazyCheckPoint(const Serializable *global_model) override { version_number_ += 1; }
int VersionNumber() const override { return version_number_; }
/*! \brief get rank of current node */