add swinv2

2023-01-11 03:35:59 +00:00 · 2023-01-11 03:35:59 +00:00 · 9ba23527b4
parent 4786b2c96b
commit 9ba23527b4
1 changed files with 2 additions and 2 deletions
--- 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
+    return model