modify pretrained model with training (#2505)

paddlex/configs/anomaly_detection/STFPM.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 1
   batch_size: 1
   learning_rate: 0.4
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/STFPM_pretrained.pdparams
+  pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null
   log_interval: 10

paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_base.yaml

@@ -20,7 +20,7 @@ Train:
   epochs_iters: 20
   batch_size: 32
   learning_rate: 0.001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_base_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_base_backbone_LAION400M_pretrained.pdparams # use CLIP_vit_base_patch16_224 pretrained
   warmup_steps: 5
   resume_path: null
   log_interval: 1

paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_large.yaml

@@ -20,7 +20,7 @@ Train:
   epochs_iters: 20
   batch_size: 32
   learning_rate: 0.001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_large_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-ShiTuV2_rec_CLIP_vit_large_backbone_LAION400M_pretrained.pdparams  # use CLIP_vit_large_patch14_224 pretrained
   warmup_steps: 5
   resume_path: null
   log_interval: 1

paddlex/configs/object_detection/PP-YOLOE_plus-L.yaml

@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-L_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-L_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

paddlex/configs/object_detection/PP-YOLOE_plus-M.yaml

@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-M_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-M_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml

@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-S_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-S_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

paddlex/configs/object_detection/PP-YOLOE_plus-X.yaml

@@ -19,7 +19,7 @@ Train:
   epochs_iters: 10
   batch_size: 8
   learning_rate: 0.0001
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-X_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-YOLOE_plus-X_obj365_pretrained.pdparams # use object365 pretrained
   warmup_steps: 100
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/Deeplabv3-R101.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R101_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R101_backbone_imagenet_pretrained.pdparams  # use ResNet101_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/Deeplabv3-R50.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R50_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3-R50_backbone_imagenet_pretrained.pdparams  # use ResNet50_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R101.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R101_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R101_backbone_imagenet_pretrained.pdparams # use ResNet101_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R50.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R50_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/Deeplabv3_Plus-R50_backbone_imagenet_pretrained.pdparams # use ResNet50_vd pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/OCRNet_HRNet-W18.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W18_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W18_backbone_imagenet_pretrained.pdparams  # use HRNet_W18 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/OCRNet_HRNet-W48.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W48_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/OCRNet_HRNet-W48_backbone_imagenet_pretrained.pdparams  # use HRNet_W48 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/PP-LiteSeg-B.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-B_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-B_backbone_imagenet_pretrained.pdparams  # use STDC2 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/PP-LiteSeg-T.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-T_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LiteSeg-T_backbone_imagenet_pretrained.pdparams  # use STDC1 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B0.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B0_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B0_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B0 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B1.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B1_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B1_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B1 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B2.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B2_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B2_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B2 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B3.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B3_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B3_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B3 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B4.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B4_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B4_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B4 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10

paddlex/configs/semantic_segmentation/SegFormer-B5.yaml

@@ -19,7 +19,7 @@ Train:
   num_classes: 2
   batch_size: 2
   learning_rate: 0.01
-  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B5_pretrained.pdparams
+  pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SegFormer-B5_backbone_imagenet_pretrained.pdparams  # use MixVisionTransformer_B5 pretrained
   warmup_steps: 0
   resume_path: null
   log_interval: 10