Learning What and Where to Transfer (ICML 2019)

Learning What and Where to Transfer (ICML 2019) https://arxiv.org/abs/1905.05901

Requirements

pytorch 1.0
CUDA 9.0
CUDNN 7

Prepare Datasets

You can download CUB-200 and Stanford Dogs datasets

CUB-200: from http://www.vision.caltech.edu/visipedia/CUB-200-2011.html
Stanford Dogs: http://vision.stanford.edu/aditya86/ImageNetDogs/

You need to run the below pre-processing script for DataLoader.

python cub200.py /data/CUB_200_2011
python dog.py /data/dog

Train L2T-ww

You can train L2T-ww models with the same settings in our paper.

python train_l2t_ww.py --dataset cub200 --datasplit cub200 --dataroot /data/CUB_200_2011
python train_l2t_ww.py --dataset dog --datasplit dog --dataroot /data/dog