@@ -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
- batch_size: 32
+ batch_size: 16
pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/CLIP_vit_large_patch14_224_pretrained.pdparams
- 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
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