Official PyTorch implementation of Domain Generalization via Rationale Invariance.
Liang Chen, Yong Zhang, Yibing Song, Anton van den Hengel, Lingqiao Liu
pip install -r requirements.txt
python -m domainbed.scripts.download --data_dir=/my/datasets/path
Environment details used for the main experiments. Every main experiment is conducted on a single NVIDIA V100 GPU.
Environment:
Python: 3.7.7
PyTorch: 1.7.1
Torchvision: 0.8.2
CUDA: 10.1
CUDNN: 7603
NumPy: 1.21.4
PIL: 7.2.0
Please refer to details in the original project page: Domainbed
The main implementation of this work is at domainbed.algorithms.RIDG
@inproceedings{chen2023domain,
title={Domain Generalization via Rationale Invariance},
author={Chen, Liang and Zhang, Yong and Song, Yibing and Hengel, Anton van den and Liu, Lingqiao},
booktitle={ICCV},
year={2023}
}
Please contact me via email (liangchen527@gmail.com) if you have any questions regarding this project.