From 9ba23527b4b7d38724ebe402187fdaea3a3b2099 Mon Sep 17 00:00:00 2001 From: zhangyubo0722 Date: Wed, 11 Jan 2023 03:35:59 +0000 Subject: [PATCH] add swinv2 --- ppcls/arch/backbone/model_zoo/swin_transformer_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py b/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py index e68ed2f90..f3944d07b 100644 --- a/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py +++ b/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py @@ -893,7 +893,7 @@ def SwinTransformerV2_base_patch4_window16_256( use_imagenet22k_pretrained=False, use_imagenet22kto1k_pretrained=False, **kwargs): - model = SwinTransformerV2( + model = SwinTransformerV2( img_size=256, embed_dim=128, depths=[2, 2, 18, 2], @@ -984,4 +984,4 @@ def SwinTransformerV2_large_patch4_window24_384( use_ssld=use_ssld, use_imagenet22k_pretrained=use_imagenet22k_pretrained, use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained) - return model \ No newline at end of file + return model