declare -a seed=(1)

for ((i=0; i<${#seed[@]}; i++));
do
python train.py \
    --model="Lin" \
    --dataset="MNIST" \
    --optimizer="ProxAdamW" \
    --epochs=500 \
    --lr=1e-3 \
    --lambda_=1e-3 \
    --max-iters=100 \
    --gamma=1e-1 \
    --seed=${seed[i]} \
    --milestones 100 200 300 400
done