From 4df192b12bbdcc4811a853024c23ec32568212de Mon Sep 17 00:00:00 2001
From: Xiang <anxiangsir@outlook.com>
Date: Sun, 1 May 2022 16:14:42 +0800
Subject: [PATCH] updated

---
 recognition/arcface_torch/README.md           | 11 ++++---
 .../arcface_torch/configs/ms1mv2_r100.py      |  2 +-
 .../arcface_torch/configs/ms1mv2_r50.py       |  2 +-
 .../arcface_torch/configs/wf12m_mbf.py        | 28 ++++++++++++++++++
 .../arcface_torch/configs/wf12m_pfc02_r100.py | 29 +++++++++++++++++++
 .../arcface_torch/configs/wf12m_r100.py       | 29 +++++++++++++++++++
 .../arcface_torch/configs/wf12m_r50.py        | 28 ++++++++++++++++++
 recognition/arcface_torch/configs/wf4m_mbf.py | 27 +++++++++++++++++
 .../arcface_torch/configs/wf4m_r100.py        | 27 +++++++++++++++++
 recognition/arcface_torch/configs/wf4m_r50.py | 27 +++++++++++++++++
 10 files changed, 204 insertions(+), 6 deletions(-)
 create mode 100644 recognition/arcface_torch/configs/wf12m_mbf.py
 create mode 100644 recognition/arcface_torch/configs/wf12m_pfc02_r100.py
 create mode 100644 recognition/arcface_torch/configs/wf12m_r100.py
 create mode 100644 recognition/arcface_torch/configs/wf12m_r50.py
 create mode 100644 recognition/arcface_torch/configs/wf4m_mbf.py
 create mode 100644 recognition/arcface_torch/configs/wf4m_r100.py
 create mode 100644 recognition/arcface_torch/configs/wf4m_r50.py

diff --git a/recognition/arcface_torch/README.md b/recognition/arcface_torch/README.md
index b49e9acc6..edd800ce0 100644
--- a/recognition/arcface_torch/README.md
+++ b/recognition/arcface_torch/README.md
@@ -66,14 +66,17 @@ globalised multi-racial testset contains 242,143 identities and 1,624,305 images
 | Datasets      | Backbone            | **MFR-ALL** | IJB-C(1E-4) | IJB-C(1E-5) | log                                                                                                                                 |
 |:--------------|:--------------------|:------------|:------------|:------------|:------------------------------------------------------------------------------------------------------------------------------------|
 | MS1MV2        | mobilefacenet-0.45G | 62.07       | 93.61       | 90.28       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv2_mbf/training.log)                     |
+| MS1MV2        | r50                 | 75.13       | 95.97       | 94.07       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv2_r50/training.log)                     |
+| MS1MV2        | r100                | 78.12       | 96.37       | 94.27       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv2_r100/training.log)                    |
 | MS1MV3        | mobilefacenet-0.45G | 63.78       | 94.23       | 91.33       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv3_mbf/training.log)                     |
-| Glint360K     | mobilefacenet-0.45G | 70.18       | 95.04       | 92.62       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/glint360k_mbf/training.log)                  |
-| MS1MV2        | r50                 | 70.35       | 95.43       | 93.34       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv2_r50/training.log)                     |
 | MS1MV3        | r50                 | 79.14       | 96.37       | 94.47       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv3_r50/training.log)                     |
-| Glint360K     | r50                 | 86.34       | 97.16       | 95.81       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/glint360k_r50/training.log)                  |
-| MS1MV2        | r100                | 69.79       | 95.85       | 93.93       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv2_r100/training.log)                    |
 | MS1MV3        | r100                | 81.97       | 96.85       | 95.02       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/ms1mv3_r100/training.log)                    |
+| Glint360K     | mobilefacenet-0.45G | 70.18       | 95.04       | 92.62       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/glint360k_mbf/training.log)                  |
+| Glint360K     | r50                 | 86.34       | 97.16       | 95.81       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/glint360k_r50/training.log)                  |
 | Glint360k     | r100                | 89.52       | 97.55       | 96.38       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/glint360k_r100/training.log)                 |
+| WF4M          | r100                | 89.87       | 97.19       | 95.48       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/wf4m_r100/training.log)                      |
+| WF12M         | r100                | 94.69       | 97.59       | 95.97       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/wf12m_r100/training.log)                     |
+| WF12M-PFC-0.2 | r100                | 94.75       | 97.60       | 95.90       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/wf12m_pfc02_r100/training.log)               |
 | WF42M-PFC-0.2 | R100                | 96.27       | 97.70       | 96.31       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/wf42m_pfc02_r100/training.log)               |
 | WF42M-PFC-0.2 | ViT-T-1.5G          | 92.04       | 97.27       | 95.68       | [click me](https://raw.githubusercontent.com/anxiangsir/insightface_arcface_log/master/wf42m_pfc02_40epoch_8gpu_vit_t/training.log) |
 
diff --git a/recognition/arcface_torch/configs/ms1mv2_r100.py b/recognition/arcface_torch/configs/ms1mv2_r100.py
index 2978d2dfe..36773489c 100644
--- a/recognition/arcface_torch/configs/ms1mv2_r100.py
+++ b/recognition/arcface_torch/configs/ms1mv2_r100.py
@@ -13,7 +13,7 @@
 config.sample_rate = 1.0
 config.fp16 = True
 config.momentum = 0.9
-config.weight_decay = 1e-4
+config.weight_decay = 5e-4
 config.batch_size = 128
 config.lr = 0.1
 config.verbose = 2000
diff --git a/recognition/arcface_torch/configs/ms1mv2_r50.py b/recognition/arcface_torch/configs/ms1mv2_r50.py
index 80d6f1168..2dab4d352 100644
--- a/recognition/arcface_torch/configs/ms1mv2_r50.py
+++ b/recognition/arcface_torch/configs/ms1mv2_r50.py
@@ -13,7 +13,7 @@
 config.sample_rate = 1.0
 config.fp16 = True
 config.momentum = 0.9
-config.weight_decay = 1e-4
+config.weight_decay = 5e-4
 config.batch_size = 128
 config.lr = 0.1
 config.verbose = 2000
diff --git a/recognition/arcface_torch/configs/wf12m_mbf.py b/recognition/arcface_torch/configs/wf12m_mbf.py
new file mode 100644
index 000000000..d1cb93b2f
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf12m_mbf.py
@@ -0,0 +1,28 @@
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "mbf"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.interclass_filtering_threshold = 0
+config.fp16 = True
+config.weight_decay = 1e-4
+config.batch_size = 128
+config.optimizer = "sgd"
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace12M"
+config.num_classes = 617970
+config.num_image = 12720066
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = []
diff --git a/recognition/arcface_torch/configs/wf12m_pfc02_r100.py b/recognition/arcface_torch/configs/wf12m_pfc02_r100.py
new file mode 100644
index 000000000..1062b876e
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf12m_pfc02_r100.py
@@ -0,0 +1,29 @@
+
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "r100"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 0.2
+config.interclass_filtering_threshold = 0
+config.fp16 = True
+config.weight_decay = 5e-4
+config.batch_size = 128
+config.optimizer = "sgd"
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace12M"
+config.num_classes = 617970
+config.num_image = 12720066
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = []
diff --git a/recognition/arcface_torch/configs/wf12m_r100.py b/recognition/arcface_torch/configs/wf12m_r100.py
new file mode 100644
index 000000000..65bfa1be4
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf12m_r100.py
@@ -0,0 +1,29 @@
+
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "r100"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.interclass_filtering_threshold = 0
+config.fp16 = True
+config.weight_decay = 5e-4
+config.batch_size = 128
+config.optimizer = "sgd"
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace12M"
+config.num_classes = 617970
+config.num_image = 12720066
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = []
diff --git a/recognition/arcface_torch/configs/wf12m_r50.py b/recognition/arcface_torch/configs/wf12m_r50.py
new file mode 100644
index 000000000..2a7284663
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf12m_r50.py
@@ -0,0 +1,28 @@
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "r50"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.interclass_filtering_threshold = 0
+config.fp16 = True
+config.weight_decay = 5e-4
+config.batch_size = 128
+config.optimizer = "sgd"
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace12M"
+config.num_classes = 617970
+config.num_image = 12720066
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = []
diff --git a/recognition/arcface_torch/configs/wf4m_mbf.py b/recognition/arcface_torch/configs/wf4m_mbf.py
new file mode 100644
index 000000000..2550f5a63
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf4m_mbf.py
@@ -0,0 +1,27 @@
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "mbf"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.fp16 = True
+config.momentum = 0.9
+config.weight_decay = 1e-4
+config.batch_size = 128
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace4M"
+config.num_classes = 205990
+config.num_image = 4235242
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = ['lfw', 'cfp_fp', "agedb_30"]
diff --git a/recognition/arcface_torch/configs/wf4m_r100.py b/recognition/arcface_torch/configs/wf4m_r100.py
new file mode 100644
index 000000000..7e95e7833
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf4m_r100.py
@@ -0,0 +1,27 @@
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "r100"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.fp16 = True
+config.momentum = 0.9
+config.weight_decay = 5e-4
+config.batch_size = 128
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace4M"
+config.num_classes = 205990
+config.num_image = 4235242
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = ['lfw', 'cfp_fp', "agedb_30"]
diff --git a/recognition/arcface_torch/configs/wf4m_r50.py b/recognition/arcface_torch/configs/wf4m_r50.py
new file mode 100644
index 000000000..b3eb0d84c
--- /dev/null
+++ b/recognition/arcface_torch/configs/wf4m_r50.py
@@ -0,0 +1,27 @@
+from easydict import EasyDict as edict
+
+# make training faster
+# our RAM is 256G
+# mount -t tmpfs -o size=140G  tmpfs /train_tmp
+
+config = edict()
+config.margin_list = (1.0, 0.0, 0.4)
+config.network = "r50"
+config.resume = False
+config.output = None
+config.embedding_size = 512
+config.sample_rate = 1.0
+config.fp16 = True
+config.momentum = 0.9
+config.weight_decay = 5e-4
+config.batch_size = 128
+config.lr = 0.1
+config.verbose = 2000
+config.dali = False
+
+config.rec = "/train_tmp/WebFace4M"
+config.num_classes = 205990
+config.num_image = 4235242
+config.num_epoch = 20
+config.warmup_epoch = 0
+config.val_targets = ['lfw', 'cfp_fp', "agedb_30"]