Home

Awesome

This is a pytorch implementation of the paper Unsupervised Domain Adaptation by Backpropagation

Environment

Network Structure

p8KTyD.md.jpg

Dataset

First, download target dataset mnist_m from pan.quark.com or Google Drive, and put mnist_m dataset into dataset/mnist_m, the structure is as follows:

--dataset--mnist_m--mnist_m_train
                 |--mnist_m_test
                 |--mnist_m_train_labels.txt
                 |--mnist_m_test_labels.txt
                 |--.gitkeep

Training

Then, run python main.py

Docker

docker build -t pytorch_dann .
docker run -it --runtime=nvidia \
  -u $(id -u):$(id -g) \
  -v /YOUR/DANN/PROJECT/dataset:/DANN/dataset \
  -v /YOUR/DANN/PROJECT/models:/DANN/models \
  pytorch_dann:latest \
  python main.py