ソースを参照

add pretrained weights address

leo-q8 8 ヶ月 前
コミット
f6de13603b

+ 1 - 0
paddlex/repo_apis/PaddleSeg_api/configs/MaskFormer_small.yaml

@@ -50,6 +50,7 @@ model:
   num_classes: 150
   backbone:
     type: SwinTransformer_small_patch4_window7_224_maskformer
+  pretrained: https://bj.bcebos.com/paddleseg/paddleseg/dygraph/ade20k/maskformer_ade20k_swin_small/pretrain/model.pdparams
 
 optimizer:
   type: AdamW

+ 1 - 0
paddlex/repo_apis/PaddleSeg_api/configs/MaskFormer_tiny.yaml

@@ -50,6 +50,7 @@ model:
   num_classes: 150
   backbone:
     type: SwinTransformer_tiny_patch4_window7_224_maskformer
+  pretrained: https://bj.bcebos.com/paddleseg/paddleseg/dygraph/ade20k/maskformer_ade20k_swin_tiny/pretrain/model.pdparams
 
 optimizer:
   type: AdamW