Update scheduler.py

aidecentralized · Jun 12, 2024 · ade4564 · ade4564
1 parent c5318c7
commit ade4564
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/scheduler.py b/src/scheduler.py
@@ -76,8 +76,8 @@ def merge_configs(self):
             "dset": "cifar10",
             "dump_dir": "./expt_dump/",
             "dpath": self.sys_config["dataset_path"] + "cifar10",
-            "num_clients": self.sys_config["dataset_splits"]["iid"]["num_clients"],
-            "samples_per_client": self.sys_config["dataset_splits"]["iid"]["samples_per_client"],
+            "num_users": self.sys_config["num_users"],
+            "samples_per_user": self.sys_config["dataset_splits"]["iid"]["samples_per_user"],
             "device_ids": get_device_ids("iid_dispfl")
         })