Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
RuiShu committed Mar 14, 2018
1 parent bd5521b commit ba1d48d
Showing 1 changed file with 33 additions and 3 deletions.
36 changes: 33 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,4 +1,34 @@
# dirt-t
A DIRT-T Approach to Unsupervised Domain Adaptation (ICLR 2018)
# DIRT-T

Implementation of [A DIRT-T Approach to Unsupervised Domain Adaptation (ICLR 2018)](https://arxiv.org/abs/1802.08735).

### Dependencies

```python
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 500
```

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

### Tensorboard Visualization

A note about MNIST data set: I accidentally divided images by 256 instead of 255. This will cause some differences in how the the images behave upon reshaping. Just keep that in mind.

0 comments on commit ba1d48d

Please sign in to comment.