-
Notifications
You must be signed in to change notification settings - Fork 1
/
personalized_setup.sh
40 lines (30 loc) · 1.26 KB
/
personalized_setup.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
NUM_CLIENTS=10
#Architecture and Dataset
MODEL=resnet18
DATASET=cifar10
#Training epochs and communication rounds
TOTALEPOCHS=60
LOCALEPOCHS=2
FINETUNE_EPOCHS=4
#Augmentations used in federated training and personalization
AUG_METHOD=nominal
FINE_TUNE_AUG_METHOD=pixel_perturbations
#Certification Arguments
SIGMA=0.12
SKIP=2
MAX=10000
#Training Hyperparameters
LR=0.1
STEP_SZ=30
BATCH_SZ=64
#Do not change anything onwards
##############################################################################################
##############################################################################################
##############################################################################################
EXP_NAME=federated-$AUG_METHOD-sigma-$SIGMA-$NUM_CLIENTS-clients-finetune-$FINE_TUNE_AUG_METHOD
CHECKPOINT=output/$EXP_NAME
JOB_NAME=Initialization-$EXP_NAME
echo $JOB_NAME
sbatch --job-name=${JOB_NAME} \
--export=ALL,MODEL=$MODEL,DATASET=$DATASET,TOTALEPOCHS=${TOTALEPOCHS},LOCALEPOCHS=${LOCALEPOCHS},AUG_METHOD=$AUG_METHOD,SIGMA=$SIGMA,EXP_NAME=$EXP_NAME,CHECKPOINT=$CHECKPOINT,NUM_CLIENTS=$NUM_CLIENTS,LR=$LR,STEP_SZ=$STEP_SZ,BATCH_SZ=$BATCH_SZ,SKIP=$SKIP,MAX=$MAX,FINETUNE_EPOCHS=$FINETUNE_EPOCHS,FINE_TUNE_AUG_METHOD=$FINE_TUNE_AUG_METHOD \
personalized_initialize.sh