glue-factory-custom/run.sh

13 lines
450 B
Bash

#python -m gluefactory.train sp+lg_homography \
# --conf gluefactory/configs/superpoint+lightglue_homography.yaml \
# data.batch_size=32 # for 1x 1080 GPU
#python -m gluefactory.train satellites_aug_both_40 \
# --conf gluefactory/configs/satellites3.yaml \
# data.batch_size=4 # for 1x 1080 GPU
python -m gluefactory.train drone2sat_v1 \
--conf gluefactory/configs/drone2sat_v1.yaml \
data.batch_size=6 # for 1x 1080 GPU