From 0519d14a061a0ecc3ed1a5bfdc31cccd149e4708 Mon Sep 17 00:00:00 2001
From: seuyou <3463423099@qq.com>
Date: Fri, 11 Mar 2022 14:53:37 +0800
Subject: [PATCH 1/3] [Feature]: Add MAE metafile

---
 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..ad62ef9bb
--- /dev/null
+++ b/configs/selfsup/mae/metafile.yaml
@@ -0,0 +1,27 @@
+Collections:
+  - Name: MAE
+    Metadata:
+      Training Data: ImageNet-1k
+      Training Techniques:
+        - LARS
+      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
\ No newline at end of file
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

From f6bfb261e82cfce6fabe9b02e6fb67e2cb332d63 Mon Sep 17 00:00:00 2001
From: seuyou <3463423099@qq.com>
Date: Fri, 11 Mar 2022 14:56:20 +0800
Subject: [PATCH 2/3] [Fix]: Fix lint

---
 configs/selfsup/mae/metafile.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/selfsup/mae/metafile.yaml b/configs/selfsup/mae/metafile.yaml
index ad62ef9bb..b49df6085 100644
--- a/configs/selfsup/mae/metafile.yaml
+++ b/configs/selfsup/mae/metafile.yaml
@@ -24,4 +24,4 @@ Models:
         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
\ No newline at end of file
+    Weights: https://download.openmmlab.com/mmselfsup/mae/mae_vit-base-p16_8xb512-coslr-400e_in1k-224_20220223-85be947b.pth

From c2205b30f47a00d0b42ec439503298ef3ec5c047 Mon Sep 17 00:00:00 2001
From: seuyou <3463423099@qq.com>
Date: Mon, 14 Mar 2022 10:14:00 +0800
Subject: [PATCH 3/3] [Fix]: Change LARS to AdamW in the metafile of MAE

---
 configs/selfsup/mae/metafile.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/selfsup/mae/metafile.yaml b/configs/selfsup/mae/metafile.yaml
index b49df6085..dc0773b02 100644
--- a/configs/selfsup/mae/metafile.yaml
+++ b/configs/selfsup/mae/metafile.yaml
@@ -3,7 +3,7 @@ Collections:
     Metadata:
       Training Data: ImageNet-1k
       Training Techniques:
-        - LARS
+        - AdamW
       Training Resources: 8x A100-80G GPUs
       Architecture:
         - ViT