Modifying the hyp-param values

Signed-off-by: achiverram28 <[email protected]>
adap · Oct 26, 2023 · 27835e7 · 27835e7
1 parent c900476
commit 27835e7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/baselines/pFedHN/pFedHN/conf/base.yaml b/baselines/pFedHN/pFedHN/conf/base.yaml
@@ -12,7 +12,6 @@ model:
   lr: 1e-2
   in_channels: 3 # 3 for CIFAR10 and CIFAR100, 1 for MNIST
   out_dim: 10 # 10 for MNIST and CIFAR10, 100 for CIFAR100
-  we_dec: 1e-3
   local: False # True: for pFedHNPC, False: pFedHN
   variant: 0 # 0: pFedHN , 1: pFedHNPC
   n_hidden: 3 # number of hidden layers for the hypernetwork: 3 for CIFAR10 and CIFAR100, 1 for MNIST

diff --git a/baselines/pFedHN/pFedHN/main.py b/baselines/pFedHN/pFedHN/main.py
@@ -45,7 +45,7 @@ def main(cfg: DictConfig):
                 node_local_layers[i].parameters(),
                 lr=cfg.model.inner_lr,
                 momentum=0.9,
-                weight_decay=cfg.model.we_dec,
+                weight_decay=cfg.model.wd,
             )
             for i in range(cfg.client.num_nodes)
         ]