Code for the paper Generalizing to Unseen Domains via Adversarial Data Augmentation
model.py: to build tf's graph
trainOps.py: to train/test
exp_configuration: config file with the hyperparameters
Python 2.7, Tensorflow 1.6.0
To obtain MNIST and SVHN dataset, run
mkdir data
python download_and_process_mnist.py
sh download_svhn.sh
To train the model, run
sh run_exp.sh GPU_IDX
where GPU_IDX is the index of the GPU to be used.
If you are interested in the topic, you might also be interested in this (related repo here)