From 827a216155d86afa54a9feb8cd5e078c18adb75b Mon Sep 17 00:00:00 2001 From: mstwutao <43307219+mstwutao@users.noreply.github.com> Date: Tue, 8 Aug 2023 02:38:18 -0500 Subject: [PATCH] [Fix] Fix typo MIMHIVIT to MAEHiViT (#1749) * fix typo MIMHIVIT to MAEHiViT * fix typo MIMHiViT to MAEHiViT --- configs/_base_/models/mae_hivit-base-p16.py | 2 +- configs/mae/mae_hivit-large-p16_8xb512-amp-coslr-400e_in1k.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/_base_/models/mae_hivit-base-p16.py b/configs/_base_/models/mae_hivit-base-p16.py index ff823ee6eb9..bac073c8401 100644 --- 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 index 0aec82eff46..2c6c47d08fd 100644 --- 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