Code and data for Learning Stable Classifiers by Transferring Unstable Features ICML 2022
Setup conda
- Create a new conda environment with the required dependencies
conda env create --file environment.yml - Activate the environment
conda activate tofu
Prepare the datasets
unzip data in datasets/mnist/MNIST/processed_data
Run experiments
sh bin/aai_mnist.sh
Save model
In ./save