File size: 1,029 Bytes
404d2af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
#!/bin/bash -l
SCRIPTPATH=$(dirname $(readlink -f "$0"))
PROJECT_DIR="${SCRIPTPATH}/../../"
# conda activate loftr
export PYTHONPATH=$PROJECT_DIR:$PYTHONPATH
cd $PROJECT_DIR
TRAIN_IMG_SIZE=832
data_cfg_path="configs/data/megadepth_trainval_${TRAIN_IMG_SIZE}.py"
main_cfg_path="configs/aspan/outdoor/aspan_train.py"
n_nodes=1
n_gpus_per_node=8
torch_num_workers=8
batch_size=1
pin_memory=true
exp_name="outdoor-ds-aspan-${TRAIN_IMG_SIZE}-bs=$(($n_gpus_per_node * $n_nodes * $batch_size))"
CUDA_VISIBLE_DEVICES='0,1,2,3,4,5,6,7' python -u ./train.py \
${data_cfg_path} \
${main_cfg_path} \
--exp_name=${exp_name} \
--gpus=${n_gpus_per_node} --num_nodes=${n_nodes} --accelerator="ddp" \
--batch_size=${batch_size} --num_workers=${torch_num_workers} --pin_memory=${pin_memory} \
--check_val_every_n_epoch=1 \
--log_every_n_steps=100 \
--flush_logs_every_n_steps=100 \
--limit_val_batches=1. \
--num_sanity_val_steps=10 \
--benchmark=True \
--max_epochs=30 \
--mode integrated
|