Merge branch 'open-mmlab:main' into main

2023-08-09 11:14:50 +08:00 · 2023-08-09 11:14:50 +08:00 · 63c7d9bcd3
parent ec3a59d23e 827a216155
commit 63c7d9bcd3
2 changed files with 2 additions and 2 deletions
--- a/configs/_base_/models/mae_hivit-base-p16.py
+++ b/configs/_base_/models/mae_hivit-base-p16.py
@ -2,7 +2,7 @@
 model = dict(
    type='MAE',
    backbone=dict(
-        type='MIMHiViT', patch_size=16, arch='base', mask_ratio=0.75),
+        type='MAEHiViT', patch_size=16, arch='base', mask_ratio=0.75),
    neck=dict(
        type='MAEPretrainDecoder',
        patch_size=16,
--- a/configs/mae/mae_hivit-large-p16_8xb512-amp-coslr-400e_in1k.py
+++ b/configs/mae/mae_hivit-large-p16_8xb512-amp-coslr-400e_in1k.py
@ -6,7 +6,7 @@ _base_ = [

 # model settings
 model = dict(
-    backbone=dict(type='MIMHiViT', arch='large'),
+    backbone=dict(type='MAEHiViT', arch='large'),
    neck=dict(type='MAEPretrainDecoder', embed_dim=768))

 # optimizer wrapper