Skip to content

Commit

Permalink
[Feature]: Add MAE metafile (#238)
Browse files Browse the repository at this point in the history
* [Feature]: Add MAE metafile

* [Fix]: Fix lint

* [Fix]: Change LARS to AdamW in the metafile of MAE
  • Loading branch information
YuanLiuuuuuu committed Mar 31, 2022
1 parent 1500038 commit 227b080
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 0 deletions.
27 changes: 27 additions & 0 deletions configs/selfsup/mae/metafile.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
Collections:
- Name: MAE
Metadata:
Training Data: ImageNet-1k
Training Techniques:
- AdamW
Training Resources: 8x A100-80G GPUs
Architecture:
- ViT
Paper:
URL: https://arxiv.org/abs/2111.06377
Title: "Masked Autoencoders Are Scalable Vision Learners"
README: configs/selfsup/mae/README.md

Models:
- Name: mae_vit-base-p16_8xb512-coslr-400e_in1k
In Collection: MAE
Metadata:
Epochs: 400
Batch Size: 4096
Results:
- Task: Self-Supervised Image Classification
Dataset: ImageNet-1k
Metrics:
Top 1 Accuracy: 83.1
Config: configs/selfsup/mae/mae_vit-base-p16_8xb512-coslr-400e_in1k.py
Weights: https://download.openmmlab.com/mmselfsup/mae/mae_vit-base-p16_8xb512-coslr-400e_in1k-224_20220223-85be947b.pth
1 change: 1 addition & 0 deletions model-index.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,3 +12,4 @@ Import:
- configs/selfsup/simclr/metafile.yml
- configs/selfsup/simsiam/metafile.yml
- configs/selfsup/swav/metafile.yml
- configs/selfsup/mae/metafile.yaml

0 comments on commit 227b080

Please sign in to comment.