Bläddra i källkod

update bs for some models

zhouchangda 1 år sedan
förälder
incheckning
413748434f

+ 1 - 1
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

+ 1 - 1
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

+ 1 - 1
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

+ 1 - 1
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