yocabon commited on
Commit
6d6d485
1 Parent(s): 67caf68

update submodule

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. dust3r +1 -1
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 293a06cbc2a255d59d6deca70a6764a1b58bb9df
 
1
+ Subproject commit 8cc725dd11a9b7371bfca37994f8585ca78b42e5