Update fedavg_fashion_cnn_noniid.yaml to use DirichletPartitioner

lishenghui · May 2, 2024 · cff545e · cff545e
1 parent 2c9b0f9
commit cff545e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/blades/tuned_examples/fedavg_fashion_cnn_noniid.yaml b/blades/tuned_examples/fedavg_fashion_cnn_noniid.yaml
@@ -12,10 +12,12 @@ fedavg_blades:
       type: FashionMNIST
       num_clients: 60
       train_batch_size: 64
-      iid:
-        grid_search: [False]
-      alpha:
-         grid_search: [10]
+
+      partitioner_config:
+        type: DirichletPartitioner
+        alpha:
+          grid_search: [10]
+        num_clients: 60
 
     evaluation_interval: 50