From 227b080c86588c52be70a0cdd4bb9273da45ecd6 Mon Sep 17 00:00:00 2001
From: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Date: Mon, 14 Mar 2022 13:14:36 +0800
Subject: [PATCH] [Feature]: Add MAE metafile (#238)

* [Feature]: Add MAE metafile

* [Fix]: Fix lint

* [Fix]: Change LARS to AdamW in the metafile of MAE
---
 configs/selfsup/mae/metafile.yaml | 27 +++++++++++++++++++++++++++
 model-index.yml                   |  1 +
 2 files changed, 28 insertions(+)
 create mode 100644 configs/selfsup/mae/metafile.yaml

diff --git a/configs/selfsup/mae/metafile.yaml b/configs/selfsup/mae/metafile.yaml
new file mode 100644
index 000000000..dc0773b02
--- /dev/null
+++ b/configs/selfsup/mae/metafile.yaml
@@ -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
diff --git a/model-index.yml b/model-index.yml
index a6106ef6f..431255cbc 100644
--- a/model-index.yml
+++ b/model-index.yml
@@ -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