add swinV1 22k weights

0a561208 · weixin_46524038 · cuicheng01 · 3672d1f2 · 0a561208
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/arch/backbone/legendary_models/swin_transformer.py ppcls/arch/backbone/legendary_models/swin_transformer.py +1 -1

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/swin_transformer.py
+++ b/ppcls/arch/backbone/legendary_models/swin_transformer.py
@@ -708,7 +708,7 @@ class SwinTransformer(TheseusLayer):
                 img_size=224,
                 patch_size=4,
                 in_chans=3,
-                 class_num=5,
+                 class_num=1000,
                 embed_dim=96,
                 depths=[2, 2, 6, 2],
                 num_heads=[3, 6, 12, 24],