init README file

README.md

@@ -1 +1,22 @@
-# pFedHN
+## Personalized Federated Learning using Hypernetworks
+
+#### Installation
+- Create a virtual environment with conda/virtualenv
+- Clone the repo
+- Run: ```cd <PATH_TO_THE CLONED REPO>```
+- Run: ```pip install -e .``` to install necessary packages and path links.
+
+---------
+
+#### Reproduce Paper Results
+
+---------
+##### PfedHN Results on CIFAR10
+- Run: ```cd experiments/pfedhn```
+- Run: ```python trainer.py```
+
+---------
+
+##### PfedHN-PC Results on CIFAR10
+- Run: ```cd experiments/pfedhn_pc```
+- Run: ```python trainer.py```

experiments/pfedhn_pc/trainer.py

@@ -264,7 +264,7 @@ def train(data_name: str, data_path: str, classes_per_node: int, num_nodes: int,
     parser.add_argument(
         "--data-name", type=str, default="cifar10", choices=['cifar10', 'cifar100'], help="data name"
     )
-    parser.add_argument("--data-path", type=str, default='/cortex/data/images', help='data path')
+    parser.add_argument("--data-path", type=str, default='data', help='data path')
     parser.add_argument("--num-nodes", type=int, default=50)
 
     ##################################