From 0a5612085546a0634bad66ffdb76232906962bf3 Mon Sep 17 00:00:00 2001
From: zhangyubo0722 <zhangyubo0722@163.com>
Date: Tue, 17 Jan 2023 11:32:20 +0000
Subject: [PATCH] add swinV1 22k weights

---
 ppcls/arch/backbone/legendary_models/swin_transformer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/arch/backbone/legendary_models/swin_transformer.py b/ppcls/arch/backbone/legendary_models/swin_transformer.py
index 9b114237..6a57ca4e 100644
--- 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],
-- 
GitLab