Discriminability and Transferability Estimation
This repository is a PyTorch implementation of the paper.
- Manually download the datasets Office, Office-Home, Office-Caltech from the official websites.
- Move
gen_list.py
inside data directory. - Generate '.txt' file for each dataset using
gen_list.py
(change dataset argument in the file accordingly).
- Train source models (shown here for Office with source A)
python train_source.py --dset office --s 0 --max_epoch 100 --trte val --gpu_id 0 --output ckps/source/
- Adapt to target (shown here for Office with target D) based on SHOT
python adapt_multi_DATE_SHOT.py --dset office --t 1 --max_epoch 15 --gpu_id 0 --output_src ckps/source/ --output ckps/adapt