From 23b0e1388c46b2fb62e549b11d1c02a0de73f4b5 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 13 Jun 2022 08:26:36 +0000
Subject: [PATCH] fix: fix bs lr

---
 .../configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml b/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml
index 2b536e1b..9ef4beb7 100644
--- a/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml
+++ b/ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml
@@ -36,7 +36,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 0.4
+    learning_rate: 0.08
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -66,11 +66,11 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 64
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 4
+      num_workers: 8
       use_shared_memory: True
 
   Eval:
-- 
GitLab