From c2802b90aae1b08dc498539e6f8f4cb59b431724 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Sat, 6 May 2023 07:23:40 +0000
Subject: [PATCH] increase bs, num_workers to speed up

---
 .../MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml         | 8 ++++----
 ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ultra.yaml    | 8 ++++----
 ppcls/configs/ImageNet/ResNet/ResNet50_ultra.yaml         | 8 ++++----
 3 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml
index f59ef3775..08c0b5119 100644
--- a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml
+++ b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml
@@ -43,7 +43,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 1.3
+    learning_rate: 7.8
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -73,11 +73,11 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 512
+      batch_size: 1536
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 4
+      num_workers: 20
       use_shared_memory: True
 
   Eval:
@@ -100,7 +100,7 @@ DataLoader:
             order: ''
     sampler:
       name: DistributedBatchSampler
-      batch_size: 64
+      batch_size: 256
       drop_last: False
       shuffle: False
     loader:
diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ultra.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ultra.yaml
index 6f942e2e5..09818a8f0 100644
--- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ultra.yaml
+++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ultra.yaml
@@ -43,7 +43,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 0.8
+    learning_rate: 3.2
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -73,11 +73,11 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 512
+      batch_size: 1024
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 4
+      num_workers: 16
       use_shared_memory: True
 
   Eval:
@@ -100,7 +100,7 @@ DataLoader:
             order: ''
     sampler:
       name: DistributedBatchSampler
-      batch_size: 64
+      batch_size: 256
       drop_last: False
       shuffle: False
     loader:
diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_ultra.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_ultra.yaml
index d5cf5a6c7..a94654f36 100644
--- a/ppcls/configs/ImageNet/ResNet/ResNet50_ultra.yaml
+++ b/ppcls/configs/ImageNet/ResNet/ResNet50_ultra.yaml
@@ -44,9 +44,9 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Piecewise
-    learning_rate: 0.1
+    learning_rate: 0.8
     decay_epochs: [30, 60, 90]
-    values: [0.1, 0.01, 0.001, 0.0001]
+    values: [0.8, 0.08, 0.008, 0.0008]
   regularizer:
     name: 'L2'
     coeff: 0.0001
@@ -75,11 +75,11 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 64
+      batch_size: 256
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 4
+      num_workers: 16
       use_shared_memory: True
 
   Eval: