update batch_size of ConvNeXt

paddlex/configs/image_classification/ConvNeXt_base_224.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 128
-  learning_rate: 0.004
+  batch_size: 32
+  learning_rate: 0.002
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/ConvNeXt_base_384.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 128
-  learning_rate: 0.004
+  batch_size: 32
+  learning_rate: 0.002
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/ConvNeXt_large_224.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 128
-  learning_rate: 0.004
+  batch_size: 16
+  learning_rate: 0.001
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/ConvNeXt_large_384.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 128
-  learning_rate: 0.004
+  batch_size: 16
+  learning_rate: 0.001
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/ConvNeXt_small.yaml

@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 128
+  batch_size: 64
   learning_rate: 0.004
   pretrain_weight_path: null
   warmup_steps: 5

paddlex/configs/image_classification/ConvNeXt_tiny.yaml

@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 64
+  batch_size: 32
   learning_rate: 0.003
   pretrain_weight_path: null
   warmup_steps: 5
@@ -35,4 +35,4 @@ Predict:
   input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg"
   kernel_option:
     run_mode: paddle
-    batch_size: 1
+    batch_size: 1

paddlex/configs/image_classification/SwinTransformer_base_patch4_window12_384.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 32
-  learning_rate: 0.003
+  batch_size: 8
+  learning_rate: 0.0015
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/SwinTransformer_base_patch4_window7_224.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 64
-  learning_rate: 0.003
+  batch_size: 32
+  learning_rate: 0.0015
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null
@@ -35,4 +35,4 @@ Predict:
   input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg"
   kernel_option:
     run_mode: paddle
-    batch_size: 1
+    batch_size: 1

paddlex/configs/image_classification/SwinTransformer_large_patch4_window12_384.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 32
-  learning_rate: 0.003
+  batch_size: 8
+  learning_rate: 0.0015
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null

paddlex/configs/image_classification/SwinTransformer_large_patch4_window7_224.yaml

@@ -17,8 +17,8 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 64
-  learning_rate: 0.003
+  batch_size: 8
+  learning_rate: 0.0015
   pretrain_weight_path: null
   warmup_steps: 5
   resume_path: null