update bs for some models

PaddlePaddle · Nov 18, 2024 · 4137484 · 4137484
1 parent 7e680c7
commit 4137484
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/paddlex/configs/image_classification/CLIP_vit_base_patch16_224.yaml b/paddlex/configs/image_classification/CLIP_vit_base_patch16_224.yaml
@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 64
+  batch_size: 32
   learning_rate: 0.0003
   pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/CLIP_vit_base_patch16_224_pretrained.pdparams
   warmup_steps: 5

diff --git a/paddlex/configs/image_classification/CLIP_vit_large_patch14_224.yaml b/paddlex/configs/image_classification/CLIP_vit_large_patch14_224.yaml
@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 32
+  batch_size: 16
   learning_rate: 0.0003
   pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/CLIP_vit_large_patch14_224_pretrained.pdparams
   warmup_steps: 5

diff --git a/paddlex/configs/image_classification/ConvNeXt_large_384.yaml b/paddlex/configs/image_classification/ConvNeXt_large_384.yaml
@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 102
   epochs_iters: 20
-  batch_size: 16
+  batch_size: 8
   learning_rate: 0.001
   pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/ConvNeXt_large_384_pretrained.pdparams
   warmup_steps: 5

diff --git a/paddlex/configs/object_detection/YOLOX-X.yaml b/paddlex/configs/object_detection/YOLOX-X.yaml
@@ -17,7 +17,7 @@ CheckDataset:
 Train:
   num_classes: 4
   epochs_iters: 10
-  batch_size: 8
+  batch_size: 4
   learning_rate: 0.01
   pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/YOLOX-X_pretrained.pdparams
   warmup_steps: 100