fixed

Author: anxiangsir

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

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

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

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