Skip to content

RuiShu/dirt-t

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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