Skip to content
/ dirt-t Public

A DIRT-T Approach to Unsupervised Domain Adaptation (ICLR 2018)

License

Notifications You must be signed in to change notification settings

RuiShu/dirt-t

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DIRT-T

Implementation of A DIRT-T Approach to Unsupervised Domain Adaptation (ICLR 2018).

Dependencies

numpy==1.14.1
scikit_image==0.13.1
scipy==1.0.0
tensorflow_gpu==1.6.0
tensorbayes==0.4.0

Download Data

Download scripts for MNIST and SVHN provided in ./data/.

Run code

Run VADA

python run_dirtt.py --datadir data --run 0 --src mnist --trg svhn --dirt 0

Run DIRT-T (pre-condition: run VADA first)

python run_dirtt.py --datadir data --run 0 --src mnist --trg svhn --dirt 5000

Tensorboard logs will be saved to ./log/ by default.

VADA and DIRT-T Performance

Test run of a single VADA run on MNIST -> SVHN, and using the final VADA model as initialization for 4 separate DIRT-T runs. DIRT-T has higher variance but, on expectation, improves upon VADA.

Tensorboard Visualization

VADA

About

A DIRT-T Approach to Unsupervised Domain Adaptation (ICLR 2018)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published