Skip to content

Latest commit

 

History

History
executable file
·
29 lines (20 loc) · 601 Bytes

README.md

File metadata and controls

executable file
·
29 lines (20 loc) · 601 Bytes
Raw

SaliencyI2PLoc

Installation

cd scripts
sh install.sh

You may required to change the coding manner of sh files using sed -i "s/\r//" *.sh to avoid the file unrecognition.

  • both pytorch1.13.1-cuda11.6 and pytorch2.1.2-cuda12.1 works

Train

cd scripts
sh train.sh # single machine
#sh train_dist.sh # DDP mode

The configuration information will be loaded all in once from the CrossModalityRetrieval.yaml file, including the optimizer, scheduler, dataset, model and other information.

Test

cd scripts
sh test.sh