From 660628c1dcb713d472222a8cbb712ee9c17f1b30 Mon Sep 17 00:00:00 2001 From: zhouchangda Date: Thu, 14 Nov 2024 09:23:25 +0000 Subject: [PATCH] update bs for some models --- .../configs/image_classification/CLIP_vit_base_patch16_224.yaml | 2 +- .../image_classification/CLIP_vit_large_patch14_224.yaml | 2 +- paddlex/configs/image_classification/ConvNeXt_large_384.yaml | 2 +- paddlex/configs/object_detection/YOLOX-X.yaml | 2 +- 4 files changed, 4 insertions(+), 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 index af7ad84871..8878143316 100644 --- 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 index 05d5c3aefb..fc7cc66c35 100644 --- 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 index 815f92bcc3..b095da6826 100644 --- 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 index cadc47e108..bfa55c667f 100644 --- 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