robustness-exps
Usage:
Run setup.sh to install required files
Find Robust EB Tickets
python find_robust_vgg_eb <output_folder> <dataset: cifar10/cifar100>
Will generate un-pruned eb30, eb50, and eb70 tickets.
Retrain Robust EB Tickets using PGD-7
python retrain_pgd7_robust_vgg_eb.py <path to unpruned eb> <prune percentage> <output/log_folder> <dataset: cifar10/cifar100>
Will prune, reset initialization back to epoch 0, and re-train using PGD7
Retrain Robust EB Tickets using FAST FGSM
python retrain_fast_robust_vgg_eb.py <path to unpruned eb> <prune percentage> <data_folder> <output/log_folder> <dataset: cifar10/cifar100>
Will prune, reset initialization back to epoch 0, and re-train using FAST FGSM
TODO: copy all experiments for resnet-18