correct vit backbone in training script

YvanYin · Jun 10, 2024 · 6a58305 · 6a58305
1 parent 9273271
commit 6a58305
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/training/mono/configs/RAFTDecoder/vit.raft5.giant2.py b/training/mono/configs/RAFTDecoder/vit.raft5.giant2.py
@@ -1,5 +1,5 @@
 _base_=['../_base_/losses/all_losses.py',
-       '../_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py',
+       '../_base_/models/encoder_decoder/dino_vit_giant2_reg.dpt_raft.py',
 
        '../_base_/datasets/ddad.py',
        '../_base_/datasets/_data_base_.py',

diff --git a/training/mono/configs/RAFTDecoder/vit.raft5.large.py b/training/mono/configs/RAFTDecoder/vit.raft5.large.py
@@ -1,5 +1,5 @@
 _base_=['../_base_/losses/all_losses.py',
-       '../_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py',
+       '../_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py',
 
        '../_base_/datasets/ddad.py',
        '../_base_/datasets/_data_base_.py',