-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathrun_exp.sh
36 lines (32 loc) · 970 Bytes
/
run_exp.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
28
29
30
31
32
33
34
35
36
# Change arguments
seeds="1"
al_method="logo" # random, entropy, coreset, badge, gcnal, alfa_mix, logo
dataset="cifar10"
dd_beta="0.1"
qmode="global" # local_only
flalgo="fedavg" # fedprox scaffold
model="cnn4conv" # mobilenet resnet18
partition="dir_balance"
reset="random_init" # continue
num_users="10"
frac="1.0"
rounds="100"
local_ep="5"
for seed in $seeds
do
for al in $al_method
do
for data in $dataset
do
for beta in $dd_beta
do
for qm in $qmode
do
CUDA_VISIBLE_DEVICES=0 python main.py --seed $seed --al_method $al --fl_algo $flalgo --model $model \
--dataset $data --partition $partition --dd_beta $beta --reset $reset --query_model_mode $qm \
--num_users $num_users --frac $frac --rounds $rounds --local_ep $local_ep --query_ratio 0.05
done
done
done
done
done