diff --git a/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py b/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py index f3944d07b2d664d8eb6e9744e768e4206514114a..e68ed2f90dab82523ad67dc6eb10f9974dd92826 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 + return model \ No newline at end of file