Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
syorami authored Oct 8, 2018
1 parent 8aa1f18 commit 4581ec1
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,14 @@
A simple implementation of Deep Domain Confusion: Maximizing for Domain Invariance.
The project contains *Pytorch* code for fine-tuning *Alexnet* as well as *DDCnet* implemented according to the original paper which adds an adaptation layer into the Alexnet.
The *office31* dataset used in the paper is also used in this implementation to test the performance of fine-tuning *Alexnet* and *DDCnet* with additional linear *MMD* loss. What’s more,

# Run the work
* Run command `python alextnet_finetune.py` to fine-tune a pretrained *Alexnet* on *office31* dataset with *full-training*.
* Run command `python DDC.py` to fine-tune a pretrained *Alexnet* on *office31* dataset with *full-training*.

# Future work
- [ ] Write data loader using *down-sample* protocol mentioned in the paper instead of using *full-training* protocol.
- [ ] Considering trying a tensorflow version to see if frameworks can have a difference on final experiment results.

# Reference
Tzeng E, Hoffman J, Zhang N, et al. Deep domain confusion: Maximizing for domain invariance[J]. arXiv preprint arXiv:1412.3474, 2014.

0 comments on commit 4581ec1

Please sign in to comment.