update submodule
Browse files
README.md
CHANGED
@@ -252,7 +252,7 @@ torchrun --nproc_per_node=4 train.py \
|
|
252 |
--test_criterion "Regr3D_ScaleShiftInv(L21, norm_mode='?avg_dis', gt_scale=True, sky_loss_value=0) + -1.*MatchingLoss(APLoss(nq='torch', fp=torch.float16), negatives_padding=12288)" \
|
253 |
--pretrained "checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth" \
|
254 |
--lr 0.0001 --min_lr 1e-06 --warmup_epochs 1 --epochs 10 --batch_size 4 --accum_iter 4 \
|
255 |
-
--save_freq 1 --keep_freq 5 --eval_freq 1 \
|
256 |
--output_dir "checkpoints/mast3r_demo"
|
257 |
|
258 |
```
|
@@ -272,7 +272,7 @@ torchrun --nproc_per_node=8 train.py \
|
|
272 |
--test_criterion "Regr3D(L21, norm_mode='?avg_dis', gt_scale=True, sky_loss_value=0) + -1.*MatchingLoss(APLoss(nq='torch', fp=torch.float16), negatives_padding=12288)" \
|
273 |
--pretrained "checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth" \
|
274 |
--lr 0.0001 --min_lr 1e-06 --warmup_epochs 8 --epochs 50 --batch_size 4 --accum_iter 2 \
|
275 |
-
--save_freq 1 --keep_freq 5 --eval_freq 1 --print_freq=10 \
|
276 |
--output_dir "checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"
|
277 |
|
278 |
```
|
|
|
252 |
--test_criterion "Regr3D_ScaleShiftInv(L21, norm_mode='?avg_dis', gt_scale=True, sky_loss_value=0) + -1.*MatchingLoss(APLoss(nq='torch', fp=torch.float16), negatives_padding=12288)" \
|
253 |
--pretrained "checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth" \
|
254 |
--lr 0.0001 --min_lr 1e-06 --warmup_epochs 1 --epochs 10 --batch_size 4 --accum_iter 4 \
|
255 |
+
--save_freq 1 --keep_freq 5 --eval_freq 1 --disable_cudnn_benchmark \
|
256 |
--output_dir "checkpoints/mast3r_demo"
|
257 |
|
258 |
```
|
|
|
272 |
--test_criterion "Regr3D(L21, norm_mode='?avg_dis', gt_scale=True, sky_loss_value=0) + -1.*MatchingLoss(APLoss(nq='torch', fp=torch.float16), negatives_padding=12288)" \
|
273 |
--pretrained "checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth" \
|
274 |
--lr 0.0001 --min_lr 1e-06 --warmup_epochs 8 --epochs 50 --batch_size 4 --accum_iter 2 \
|
275 |
+
--save_freq 1 --keep_freq 5 --eval_freq 1 --print_freq=10 --disable_cudnn_benchmark \
|
276 |
--output_dir "checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"
|
277 |
|
278 |
```
|
dust3r
CHANGED
@@ -1 +1 @@
|
|
1 |
-
Subproject commit
|
|
|
1 |
+
Subproject commit 8cc725dd11a9b7371bfca37994f8585ca78b42e5
|