This repository contains codes for the training and evaluation of our CVPR-23 paper DART:Diversify-Aggregate-Repeat Training Improves Generalization of Neural Networks main and supplementary. The arxiv link for the paper is also available.
- Python 3.6.9
- PyTorch 1.8
- Torchvision 0.8.0
- Numpy 1.19.2
For training DART on Domain Generalization task:
python
@inproceedings{jain2023dart,
title={DART: Diversify-Aggregate-Repeat Training Improves Generalization of Neural Networks},
author={Jain, Samyak and Addepalli, Sravanti and Sahu, Pawan Kumar and Dey, Priyam and Babu, R Venkatesh},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={16048--16059},
year={2023}
}