#!/bin/bash

# choices of sparse_init are [snip, ERK, GraSP, ER, uniform, uniform_plus]

sparse_init=snip
data=cifar10
model=cifar_resnet_20
for seed in 18 19 20
do
    for density in 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9
    do
        python main.py --sparse --seed $seed --sparse_init $sparse_init --fix --lr 0.1 --density $density --model $model --data $data --epoch 160
    done
done
model=cifar_resnet_32
for seed in 18 19 20
do
    for density in 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9
    do
        python main.py --sparse --seed $seed --sparse_init $sparse_init --fix --lr 0.1 --density $density --model $model --data $data --epoch 160
    done
done
model=cifar_resnet_44
for seed in 18 19 20
do
    for density in 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9
    do
        python main.py --sparse --seed $seed --sparse_init $sparse_init --fix --lr 0.1 --density $density --model $model --data $data --epoch 160
    done
done
model=cifar_resnet_56
for seed in 18 19 20
do
    for density in 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9
    do
        python main.py --sparse --seed $seed --sparse_init $sparse_init --fix --lr 0.1 --density $density --model $model --data $data --epoch 160
    done
done
model=cifar_resnet_110
for seed in 18 19 20
do
    for density in 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9
    do
        python main.py --sparse --seed $seed --sparse_init $sparse_init --fix --lr 0.1 --density $density --model $model --data $data --epoch 160
    done
done




