separated the configs for object detect

aidecentralized · Aug 17, 2024 · c0593dc · c0593dc
1 parent 82468a6
commit c0593dc
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/configs/algo_config.py b/src/configs/algo_config.py
@@ -17,6 +17,18 @@
 }
 
 traditional_fl = {
+    "algo": "fedavg",
+    "exp_id": 10,
+    "exp_type": "iid_clients_federated",
+    # Learning setup
+    "epochs": 1000,
+    "model": "resnet34",
+    "model_lr": 3e-4,
+    "batch_size": 256,
+    "exp_keys": [],
+}
+
+fedavg_object_detect = {
     "algo": "fedavg",
     "exp_id": "test_modular_yolo",
     "exp_type": "test",
@@ -28,4 +40,4 @@
     "exp_keys": [],
 }
 
-current_config = traditional_fl
+current_config = fedavg_object_detect