experiment setups

edml/config/battery/resnet110_cifar100_cost.yaml

+deduction_per_second: 0.02
+deduction_per_mflop: 0.00000005
+deduction_per_mbyte_received: 0.0002
+deduction_per_mbyte_sent: 0.0002

edml/config/topology/equal_batteries_10_devices.yaml

+devices: [
+    {
+        device_id: "d0",
+        address: "localhost:50051",
+        battery_capacity: 1000000,
+        torch_device: cuda:0
+    },
+    {
+        device_id: "d1",
+        address: "localhost:50052",
+        battery_capacity: 1000000,
+        torch_device: cuda:1
+    },
+    {
+        device_id: "d2",
+        address: "localhost:50053",
+        battery_capacity: 1000000,
+        torch_device: cuda:2
+    },
+    {
+        device_id: "d3",
+        address: "localhost:50054",
+        battery_capacity: 1000000,
+        torch_device: cuda:0
+    },
+    {
+        device_id: "d4",
+        address: "localhost:50055",
+        battery_capacity: 1000000,
+        torch_device: cuda:1
+    },
+    {
+        device_id: "d5",
+        address: "localhost:50056",
+        battery_capacity: 1000000,
+        torch_device: cuda:2
+    },
+    {
+        device_id: "d6",
+        address: "localhost:50057",
+        battery_capacity: 1000000,
+        torch_device: cuda:0
+    },
+    {
+        device_id: "d7",
+        address: "localhost:50058",
+        battery_capacity: 1000000,
+        torch_device: cuda:1
+    },
+    {
+        device_id: "d8",
+        address: "localhost:50059",
+        battery_capacity: 1000000,
+        torch_device: cuda:2
+    },
+    {
+        device_id: "d9",
+        address: "localhost:50060",
+        battery_capacity: 1000000,
+        torch_device: cuda:0
+    }
+]

edml/config/topology/resnet110_cifar100_batteries.yaml

+devices: [
+  {
+    device_id: "d0",
+    address: "localhost:50051",
+    battery_capacity: 400,
+
+  },
+  {
+    device_id: "d1",
+    address: "localhost:50052",
+    battery_capacity: 400
+  },
+  {
+    device_id: "d2",
+    address: "localhost:50053",
+    battery_capacity: 300
+  },
+  {
+    device_id: "d3",
+    address: "localhost:50054",
+    battery_capacity: 200
+  },
+  {
+    device_id: "d4",
+    address: "localhost:50055",
+    battery_capacity: 200
+  }
+]

edml/core/server.py

@@ -298,7 +298,9 @@ class DeviceServer:
                 print(
                     f"\n{Fore.RED}ADAPTIVE TRESHOLD REACHED, NEXT BATCH\n{Fore.RESET}"
                 )
-                self.node_device.log({"adaptive_learning_threshold_applied": True})
+                self.node_device.log(
+                    {"adaptive_learning_threshold_applied": server_gradients.size(0)}
+                )
                 continue
 
             num_client_gradients = len(client_forward_pass_responses)