Skip to content

Latest commit

 

History

History
31 lines (21 loc) · 773 Bytes

README.md

File metadata and controls

31 lines (21 loc) · 773 Bytes

Requirements

Prepare Pre-trained Models

mkdir logs && cd logs
# download the the pretrained model and save it to logs

The file tree should be

logs
└── pretrained
    ├── resnet50_ibn_a.pth.tar
    └── resnet101_ibn_a.pth.tar

The pretrained models for ibn should be found in IBN-Net

Training

# Note, this script is for  my slurm environment 
# and you should modify it to adapt you own training environment
./scripts/multi_source/train_to_market.sh resnet_ibn50a VI_Face_1080TI 8 ${description} ${port}