A repository that implements intermediate robustness training and evaluation from the NeurIPS 2021 paper Robustness between the worst and average case. Created by Leslie Rice, Anna Bair, Huan Zhang and Zico Kolter.

• To install all required packages run: pip install -r requirements.txt.
• Pretrained model weights can be downloaded here.
• To train (l_infty perturbations), run python train.py -c {path_to_training_config_file}.json.
• To evaluate (l_infty perturbations), run python train.py -c {path_to_evaluation_config_file}.json.
• To train (spatial transformations), run python train_discrete.py -c {path_to_training_config_file}.json.
• To evaluate (spatial transformations), run python eval_discrete.py --checkpoint {path_to_model_checkpoint}.pth.