Update sweep_ss.py

FedML-AI · May 17, 2021 · f3c7a74 · f3c7a74
1 parent 7ead1d6
commit f3c7a74
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/experiments/distributed/transformer_exps/run_ss_exps/sweep_ss.py b/experiments/distributed/transformer_exps/run_ss_exps/sweep_ss.py
@@ -50,13 +50,13 @@ def wait_for_the_training_process():
 # sh run_seq2seq.sh FedProx "niid_cluster_clients=100_alpha=0.1" 1e-1 1 0.5 15
 
 hps = [
-    'FedOPT "uniform" 5e-5 1 0.5 15',
+    # 'FedOPT "uniform" 5e-5 1 0.5 15',
     # 'FedOPT "niid_cluster_clients=100_alpha=0.1" 5e-5 1 0.5 15',
     # 'FedOPT "niid_cluster_clients=100_alpha=0.01" 5e-5 1 0.5 15',
 
-    'FedAvg "uniform" 5e-5 1 0.5 15',
-    'FedAvg "niid_cluster_clients=100_alpha=0.1" 5e-5 1 0.5 15',
-    'FedAvg "niid_cluster_clients=100_alpha=0.01" 5e-5 1 0.5 15',
+    'FedAvg "uniform" 1e-1 1 0.5 15',
+    'FedAvg "niid_cluster_clients=100_alpha=0.1" 1e-1 1 0.5 15',
+    'FedAvg "niid_cluster_clients=100_alpha=0.01" 1e-1 1 0.5 15',
 
     'FedProx "uniform" 1e-1 1 0.5 15',
     'FedProx "niid_cluster_clients=100_alpha=0.1" 1e-1 1 0.5 15',