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.