Code for the paper Generalizing to Unseen Domains via Adversarial Data Augmentation
Overview
Files
model.py: to build tf's graph
trainOps.py: to train/test
exp_configuration: config file with the hyperparameters
Prerequisites
Python 2.7, Tensorflow 1.6.0
How it works
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.