mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

add swinv2

Browse Source
This commit is contained in:
zhangyubo0722 2023-01-09 11:33:46 +00:00 committed by cuicheng01
parent 74de24f621
commit 4786b2c96b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/arch/backbone/model_zoo/swin_transformer_v2.py
View File

@ -893,7 +893,7 @@ def SwinTransformerV2_base_patch4_window16_256(
use_imagenet22k_pretrained=False, use_imagenet22k_pretrained=False,
use_imagenet22kto1k_pretrained=False, use_imagenet22kto1k_pretrained=False,
**kwargs): **kwargs):
model = SwinTransformerV2( model = SwinTransformerV2(
img_size=256, img_size=256,
embed_dim=128, embed_dim=128,
depths=[2, 2, 18, 2], depths=[2, 2, 18, 2],
@ -984,4 +984,4 @@ def SwinTransformerV2_large_patch4_window24_384(
use_ssld=use_ssld, use_ssld=use_ssld,
use_imagenet22k_pretrained=use_imagenet22k_pretrained, use_imagenet22k_pretrained=use_imagenet22k_pretrained,
use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained) use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained)
return model return model