24 lines
623 B
Bash
24 lines
623 B
Bash
set -e
|
|
|
|
lda_alpha=$1
|
|
cudaid=$2
|
|
gamma=$3
|
|
lr=$4
|
|
eps=$5
|
|
alpha=$6
|
|
annealing=$7
|
|
|
|
|
|
echo $lda_alpha
|
|
echo $cudaid
|
|
echo $gamma
|
|
echo $lr
|
|
echo $eps
|
|
echo $alpha
|
|
echo $annealing
|
|
|
|
for (( i=0; i<5; i++ ))
|
|
do
|
|
CUDA_VISIBLE_DEVICES="${cudaid}" python federatedscope/main.py --cfg scripts/wide_valley_exp_scripts/fedentsgd_on_cifar10.yaml seed $i data.splitter_args "[{'alpha': ${lda_alpha}}]" trainer.local_entropy.gamma $gamma fedopt.optimizer.lr 1.0 fedopt.annealing $annealing trainer.local_entropy.eps $eps trainer.local_entropy.alpha $alpha train.optimizer.lr $lr expname fedentsgd_${lda_alpha}_${gamma}_${eps}_${annealing}_${i}
|
|
done
|