[Fix] Fix typo MIMHIVIT to MAEHiViT (#1749)

* fix typo MIMHIVIT to MAEHiViT * fix typo MIMHiViT to MAEHiViT
open-mmlab · Aug 8, 2023 · 827a216 · 827a216
1 parent 1dda91b
commit 827a216
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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,

diff --git 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 @@
 
 # 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