Преглед на файлове

add pretrained weights address

leo-q8 преди 8 месеца
родител
ревизия
f6de13603b
променени са 2 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 1 0
      paddlex/repo_apis/PaddleSeg_api/configs/MaskFormer_small.yaml
  2. 1 0
      paddlex/repo_apis/PaddleSeg_api/configs/MaskFormer_tiny.yaml

+ 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