diff --git a/src/scheduler.py b/src/scheduler.py index 62a155e..3dbb332 100644 --- 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"] })