update configs

mattoslmp · Jul 14, 2024 · 08ac210 · 08ac210
1 parent 47af038
commit 08ac210
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/eval/dense_prediction/configs/dpt/dpt_mae-b14_8xb2-160k_nyu-518x518_proteus.py b/eval/dense_prediction/configs/dpt/dpt_mae-b14_8xb2-160k_nyu-518x518_proteus.py
@@ -7,7 +7,7 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='pretrain/dinov2_vitb14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vitb14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/dpt/dpt_mae-l14_8xb2-160k_nyu-518x518_proteus.py b/eval/dense_prediction/configs/dpt/dpt_mae-l14_8xb2-160k_nyu-518x518_proteus.py
@@ -7,7 +7,7 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='pretrain/dinov2_vitl14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vitl14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/dpt/dpt_mae-s14_8xb2-160k_nyu-518x518_proteus.py b/eval/dense_prediction/configs/dpt/dpt_mae-s14_8xb2-160k_nyu-518x518_proteus.py
@@ -7,7 +7,7 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='/home/qw/yitian/mmsegmentation/pretrained/dinov2_vits14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vits14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/dpt/dpt_mae-s14_8xb2-160k_nyu-518x518_proteus_probing.py b/eval/dense_prediction/configs/dpt/dpt_mae-s14_8xb2-160k_nyu-518x518_proteus_probing.py
@@ -7,7 +7,7 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='/home/qw/yitian/mmsegmentation/pretrained/dinov2_vits14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vits14_proteus.pth',
     backbone=dict(
         type='MAE_fix',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/mae/mae-base_upernet_8xb2-amp-60k_ade20k-518x518_proteus.py b/eval/dense_prediction/configs/mae/mae-base_upernet_8xb2-amp-60k_ade20k-518x518_proteus.py
@@ -6,7 +6,7 @@
 data_preprocessor = dict(size=crop_size)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='pretrain/dinov2_vitb14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vitb14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/mae/mae-large_upernet_8xb2-amp-160k_ade20k-518x518_proteus.py b/eval/dense_prediction/configs/mae/mae-large_upernet_8xb2-amp-160k_ade20k-518x518_proteus.py
@@ -6,7 +6,7 @@
 data_preprocessor = dict(size=crop_size)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='pretrain/dinov2_vitl14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vitl14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),

diff --git a/...se_prediction/configs/mae/mae-small_linear_8xb2-amp-60k_ade20k-518x518_proteus_probing.py b/...se_prediction/configs/mae/mae-small_linear_8xb2-amp-60k_ade20k-518x518_proteus_probing.py
@@ -8,7 +8,7 @@
 model = dict(
     type='EncoderDecoder',
     data_preprocessor=data_preprocessor,
-    pretrained='/home/qw/yitian/mmsegmentation/pretrained/dinov2_vits14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vits14_proteus.pth',
     backbone=dict(
         type='MAE_fix',
         img_size=(518, 518),

diff --git a/eval/dense_prediction/configs/mae/mae-small_upernet_8xb2-amp-60k_ade20k-518x518_proteus.py b/eval/dense_prediction/configs/mae/mae-small_upernet_8xb2-amp-60k_ade20k-518x518_proteus.py
@@ -6,7 +6,7 @@
 data_preprocessor = dict(size=crop_size)
 model = dict(
     data_preprocessor=data_preprocessor,
-    pretrained='/home/qw/yitian/mmsegmentation/pretrained/dinov2_vits14_ibot_mse_aligned_old_aug_ema_wo_mixup.pth',
+    pretrained='pretrain/dinov2_vits14_proteus.pth',
     backbone=dict(
         type='MAE',
         img_size=(518, 518),