소스 검색

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