Code release for Metric-Learning-Assisted Domain Adaptation (Neurocomputing, 2021, IF: 5.779)
- All rights reserved by Yueming Yin, Email: 1018010514@njupt.edu.cn (or yinym96@qq.com).
- python 3.6+
- PyTorch 1.0
pip install -r requirements.txt
-
download datasets
-
modify your root path, task name and GPU ids
Line 38-41 in train.py
-
train:
python train.py
-
monitor (tensorboard required)
tensorboard --logdir .\log --port 8888
-
view the results of our ablation study
tensorboard --logdir .\Ablation Study --port 8888
Please cite:
@article{yin2021metric,
title={Metric-learning-assisted domain adaptation},
author={Yin, Yueming and Yang, Zhen and Hu, Haifeng and Wu, Xiaofu},
journal={Neurocomputing},
volume={454},
pages={268--279},
year={2021},
publisher={Elsevier}
}
or
Yin, Yueming, Zhen Yang, Haifeng Hu, and Xiaofu Wu. "Metric-learning-assisted domain adaptation." Neurocomputing 454 (2021): 268-279.
- 1018010514@njupt.edu.cn (Yueming Yin)