forked from haozhaowang/DaFKD2023
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_fed_standalone.sh
executable file
·50 lines (32 loc) · 981 Bytes
/
run_fed_standalone.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/env bash
GPU=0
CLIENT_NUM=5
WORKER_NUM=2
BATCH_SIZE=32
ALPHA=1
DATASET=fashion_mnist
DISTILLATION_DATASET=mnist
MODEL=DaFKD
TYPE="cv" # cv/nlp
DISTRIBUTION=hetero
ROUND=30
CLIENT_OPTIMIZER=adam
EPOCH=5
GAN_EPOCH=5
D_EPOCH=1
ED_EPOCH=30
NOISE_DIMENSION=100
TEMPERATURE=20.0
LR=0.0001
ES_LR=0.0001
AGGREGATION=FedDF
CI=0
BASELINE="DaFKD"
CUDA_VISIBLE_DEVICES=0 python -B main_fed.py --gpu $GPU --dataset $DATASET \
--model $MODEL --partition_method $DISTRIBUTION --alpha $ALPHA --dom $TYPE\
--aggregation_method $AGGREGATION --client_num_in_total $CLIENT_NUM \
--client_num_per_round $WORKER_NUM --comm_round $ROUND --epochs $EPOCH \
--client_optimizer $CLIENT_OPTIMIZER --batch_size $BATCH_SIZE \
--d_epoch $D_EPOCH --ed_epoch $ED_EPOCH --gan_epoch $GAN_EPOCH \
--noise_dimension $NOISE_DIMENSION --temperature $TEMPERATURE --lr $LR \
--es_lr $ES_LR --ci $CI --baseline $BASELINE --distillation_dataset $DISTILLATION_DATASET