From 48ecbc7b49dc60dbc7cb339bcc96f0c080a1f5cc Mon Sep 17 00:00:00 2001 From: Xiang Date: Tue, 29 Mar 2022 20:52:17 +0800 Subject: [PATCH] fixed --- recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py | 2 +- recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py | 2 +- recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py | 2 +- recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py b/recognition/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py index 13e590d69..adf21c97a 100644 --- 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 index 1174c5df3..5d35830ba 100644 --- 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 index 8c3d5d2b6..e34dd1c11 100644 --- 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 index 24ad5746a..a44a5d771 100644 --- 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