Code for "Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalization"
We place the three experiments
(anti causal colored MNIST, covariate shift colored MNIST
and invariance unit tests)
and the respective instructions
in three sub folders (AC-CMNIST, CS-CMNIST and InvarianceUnitTests).
Requirements
- torch 1.6.0
- torchvision 0.7.0
- numpy 1.19.1
- tqdm 4.41.1