Initial commit

metric_depth/dist_train.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+now=$(date +"%Y%m%d_%H%M%S")
+
+epoch=120
+bs=4
+gpus=8
+lr=0.000005
+encoder=vitl
+dataset=hypersim # vkitti
+img_size=518
+min_depth=0.001
+max_depth=20 # 80 for virtual kitti
+pretrained_from=../checkpoints/depth_anything_v2_${encoder}.pth
+save_path=exp/hypersim # exp/vkitti
+
+mkdir -p $save_path
+
+python3 -m torch.distributed.launch \
+    --nproc_per_node=$gpus \
+    --nnodes 1 \
+    --node_rank=0 \
+    --master_addr=localhost \
+    --master_port=20596 \
+    train.py --epoch $epoch --encoder $encoder --bs $bs --lr $lr --save-path $save_path --dataset $dataset \
+    --img-size $img_size --min-depth $min_depth --max-depth $max_depth --pretrained-from $pretrained_from \
+    --port 20596 2>&1 | tee -a $save_path/$now.log