# This is the code to run the reparametrized sparsity experiment, to run the experiment run:

1) Install requirements.txt

2) python harness.py --config configs/cifar10_mw_resnet20adam.yaml

Or

python harness.py --config configs/cifar10_mw_resnet20adamw.yaml


The code is made to run on 4 A100 GPUs but can be adapted by changing the config files to run on 1 or any number of GPUs by changing the batch size as it is per GPU.
Currently it is 128 x 4 = 512.