forked from Erosinho13/MLDL23-FL-project
-
Notifications
You must be signed in to change notification settings - Fork 2
/
clavbn.sh
45 lines (41 loc) · 1.01 KB
/
clavbn.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
git clone https://github.com/SamueleVanini/MLDL23-FL-project.git
# Import packages needed in colab
pip install wandb
pip install overrides
# Login in wandb
wandb login dfb691367550ee2d3f2c3a4f4849e10d065188fb
# Move in the correct folder
cd MLDL23-FL-project
python run.py \
--project exam-project \
--exp_name silo_no_fda_up_10_cl_2 \
--load_checkpoint fed-server-silo-self-learning-100.torch mldlproject/silo-self-learning/6f9npf8q \
--seed 0 \
--training_ds idda_silo \
--test_ds idda_silo \
--model deeplabv3_mobilenetv2 \
--num_epochs 1 \
--bs 4 \
--optimizer SGD \
--lr 0.01 \
--weight_decay 0 \
--momentum 0.9 \
--lr_policy poly \
--lr_power 0.9 \
--lr_decay_step 15 \
--lr_decay_factor 0.1 \
--rrc_transform \
--min_scale 0.5 \
--max_scale 2.0 \
--h_resize 512 \
--w_resize 928 \
--norm eros_norm \
--phase test \
silo_self_learning \
--num_rounds 100 \
--clients_per_round 2 \
--update_teach 10 \
--conf_threshold 0.9 \
--alpha 0.3 \
--beta 0.7 \
--tau 2