-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathrun_cifar10.sh
27 lines (25 loc) · 2.41 KB
/
run_cifar10.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
data_partition="noniid-skew"
client=10
iternum=200
alpha=0.8
beta=0.1
# Baselines
python local_training.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedavg.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedavg_ft.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedprox.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedprox_ft.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python cfl.py --gpu "5" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python perfedavg.py --gpu "6" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python pfedme.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedamp.py --gpu "6" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python ditto.py --gpu "5" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedrep.py --gpu "6" --partition $data_partition --n_parties $client --num_rep_iterations $iternum --beta $beta
python pfedhn.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python fedrod.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
python knn_per.py --gpu "7" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta
# Ours
python pfedgraph_cosine.py --gpu "0" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta --alpha $alpha --difference_measure "all"
python pfedgraph_approx.py --gpu "5" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta --alpha $alpha --difference_measure "all"
python pfedgraph_cosine.py --gpu "1" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta --alpha $alpha --difference_measure "fc"
python pfedgraph_approx.py --gpu "1" --partition $data_partition --n_parties $client --num_local_iterations $iternum --beta $beta --alpha $alpha --difference_measure "fc"