fixed

deepinsight · Mar 29, 2022 · 48ecbc7 · 48ecbc7
1 parent a2ab56d
commit 48ecbc7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py b/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py
@@ -14,7 +14,7 @@
 config.fp16 = True
 config.momentum = 0.9
 config.weight_decay = 5e-4
-config.batch_size = 512
+config.batch_size = 128
 config.lr = 0.4
 config.verbose = 2000
 config.dali = False

diff --git a/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py b/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py
@@ -14,7 +14,7 @@
 config.fp16 = True
 config.momentum = 0.9
 config.weight_decay = 5e-4
-config.batch_size = 512
+config.batch_size = 128
 config.lr = 0.4
 config.verbose = 2000
 config.dali = False

diff --git a/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py b/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py
@@ -14,7 +14,7 @@
 config.fp16 = True
 config.momentum = 0.9
 config.weight_decay = 5e-4
-config.batch_size = 512
+config.batch_size = 128
 config.lr = 0.4
 config.verbose = 2000
 config.dali = False

diff --git a/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py b/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py
@@ -14,7 +14,7 @@
 config.fp16 = True
 config.momentum = 0.9
 config.weight_decay = 5e-4
-config.batch_size = 512
+config.batch_size = 128
 config.lr = 0.4
 config.verbose = 2000
 config.dali = False