removed hardcoding from scheduler

aidecentralized · Jul 1, 2024 · f6a9f62 · f6a9f62
1 parent ee4dc39
commit f6a9f62
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/scheduler.py b/src/scheduler.py
@@ -72,10 +72,11 @@ def assign_config_by_path(self, sys_config_path, algo_config_path):
     def merge_configs(self):
         self.config = self.algo_config.copy()
         self.config.update({
-            "dset": "cifar10",
-            "dump_dir": "./expt_dump/",
-            "dpath": self.sys_config["dataset_path"] + "cifar10",
+            "dset": self.sys_config["dset"],
+            "dump_dir": self.sys_config["dump_dir"],
+            "dpath": self.sys_config["dpath"],
             "num_users": self.sys_config["num_users"],
+            "seed": self.config["seed"],
             "samples_per_user": self.sys_config["dataset_splits"]["samples_per_user"],
             "device_ids": self.sys_config["device_ids"]
         })