Эх сурвалжийг харах

update the pretrain in exporting to be consistent with the past

gaotingquan 1 жил өмнө
parent
commit
66ff6b405c

+ 2 - 2
paddlex/configs/image_classification/CLIP_vit_base_patch16_224.yaml

@@ -31,11 +31,11 @@ Evaluate:
   log_interval: 1
 
 Export:
-  weight_path: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/foundation_models/CLIP_vit_base_patch16_224.pdparams
+  weight_path: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/foundation_models/CLIP_vit_base_patch16_224_ft_in1k_pretrained.pdparams 
 
 Predict:
   model_dir: "output/best_model"
   input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg"
   kernel_option:
     run_mode: paddle
-    batch_size: 1
+    batch_size: 1

+ 2 - 2
paddlex/configs/image_classification/CLIP_vit_large_patch14_224.yaml

@@ -31,11 +31,11 @@ Evaluate:
   log_interval: 1
 
 Export:
-  weight_path: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/foundation_models/CLIP_vit_large_patch14_224.pdparams
+  weight_path: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/foundation_models/CLIP_vit_large_patch14_224_ft_in1k_pretrained.pdparams
 
 Predict:
   model_dir: "output/best_model"
   input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg"
   kernel_option:
     run_mode: paddle
-    batch_size: 1
+    batch_size: 1

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

@@ -30,11 +30,11 @@ Evaluate:
   log_interval: 10
 
 Export:
-  weight_path: https://paddleseg.bj.bcebos.com/dygraph/cityscapes/pp_liteseg_stdc1_cityscapes_1024x512_scale0.5_160k/model.pdparams
+  weight_path: https://paddleseg.bj.bcebos.com/dygraph/cityscapes/pp_liteseg_stdc1_cityscapes_1024x512_scale1.0_160k/model.pdparams
 
 Predict:
   model_dir: "output/best_model/model"
   input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_semantic_segmentation_002.png"
   kernel_option:
     run_mode: paddle
-    batch_size: 1
+    batch_size: 1