mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

* fix typo MIMHIVIT to MAEHiViT

* fix typo MIMHiViT to MAEHiViT
pull/1753/head^2
mstwutao 2023-08-08 02:38:18 -05:00 committed by GitHub
parent 1dda91bf24
commit 827a216155
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/_base_/models/mae_hivit-base-p16.py
View File

@ -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,

2
configs/mae/mae_hivit-large-p16_8xb512-amp-coslr-400e_in1k.py
View File

@ -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