conda update -n base -c defaults conda
conda create -n milkshake python==3.10
conda activate milkshake
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
python -m pip install -e .

@inproceedings{labonte23towards,
  author={Tyler LaBonte and Vidya Muthukumar and Abhishek Kumar},
  title={Towards Last-layer Retraining for Group Robustness with Fewer Annotations},
  booktitle={Conference on Neural Information Processing Systems (NeurIPS)},
  year={2023},
}