-
Notifications
You must be signed in to change notification settings - Fork 3
/
fedae_fintuning_script.sh
42 lines (29 loc) · 2.48 KB
/
fedae_fintuning_script.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
#!/bin/bash
# Local Epoch
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.01 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 10 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.01 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.01 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 30 --gpu $4
# Learning Rate
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.01 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.05 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
# Batch Size
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 64 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 128 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 256 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
# Latent dim
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 8 --batch_size 128 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 16 --batch_size 128 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4
python3 main.py --algorithm FedAE --dataset $1 --learning_rate 0.1 --num_global_iters 100 \
--clients $2 --dim 20 --batch_size 128 --subusers $3 --threshold 0.02 --local_epochs 20 --gpu $4