Fix the path to save the policy of FedEx (#141)

alibaba · Jun 5, 2022 · e2e1cc9 · e2e1cc9
1 parent 93fe045
commit e2e1cc9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/federatedscope/autotune/fedex/server.py b/federatedscope/autotune/fedex/server.py
@@ -377,8 +377,11 @@ def check_and_save(self):
 
             if self._cfg.federate.save_to != '':
                 # save the policy
-                with open(os.path.join(self._cfg.outdir, "policy.npy"),
-                          'wb') as ops:
+
+                pi_ckpt_path = self._cfg.federate.save_to[:self._cfg.federate.
+                                                          save_to.rfind(
+                                                              '.')] + "_pi.npy"
+                with open(pi_ckpt_path, 'wb') as ops:
                     np.save(ops, self._z)
 
             if self.model_num > 1: