21 lines
584 B
Bash
21 lines
584 B
Bash
set -e
|
|
|
|
cd ../..
|
|
|
|
cudaid=$1
|
|
dataset=$2
|
|
trainer=$3
|
|
mu=$4
|
|
alpha=$5
|
|
|
|
if [ ! -d "out" ];then
|
|
mkdir out
|
|
fi
|
|
|
|
for k in {1..3}
|
|
do
|
|
echo "k=${k}, Trainer=${trainer}, data=${dataset}, mu=${mu}, alpha=${alpha} starts..."
|
|
python federatedscope/main.py --cfg federatedscope/gfl/flitplus/fedalgo_cls.yaml device ${cudaid} data.type ${dataset} trainer.type ${trainer} fedprox.use True fedprox.mu ${mu} flitplus.alpha ${alpha} seed ${k} >>out/${trainer}_on_${dataset}_k${k}_mu${mu}_alpha${alpha}.log 2>&1
|
|
echo "k=${k}, Trainer=${trainer}, data=${dataset}, mu=${mu}, alpha=${alpha} ends."
|
|
done
|