fix federated learning plugin (#8027)

dmlc · Jun 24, 2022 · 0725fd6 · 0725fd6
1 parent a68580e
commit 0725fd6
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/plugin/federated/engine_federated.cc b/plugin/federated/engine_federated.cc
@@ -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 */