fix configuration bugs

ziplab · Nov 28, 2021 · e33302f · e33302f
1 parent e5a0d74
commit e33302f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/SPViT_Swin/configs/spvit_swin_tn_l28_t32_ft.yaml b/SPViT_Swin/configs/spvit_swin_tn_l28_t32_ft.yaml
@@ -3,9 +3,9 @@ MODEL:
   NAME: spvit_swin_tn_l28_t32_ft
   DROP_PATH_RATE: 0.2
   SWIN:
-    EMBED_DIM: 128
-    DEPTHS: [ 2, 2, 18, 2 ]
-    NUM_HEADS: [ 4, 8, 16, 32 ]
+    EMBED_DIM: 96
+    DEPTHS: [ 2, 2, 6, 2 ]
+    NUM_HEADS: [ 3, 6, 12, 24 ]
     WINDOW_SIZE: 7
 DATA:
   NUM_WORKERS: 10

diff --git a/SPViT_Swin/configs/spvit_swin_tn_l28_t32_search.yaml b/SPViT_Swin/configs/spvit_swin_tn_l28_t32_search.yaml
@@ -3,9 +3,9 @@ MODEL:
   NAME: spvit_swin_tn_l28_t32_search
   DROP_PATH_RATE: 0.2
   SWIN:
-    EMBED_DIM: 128
-    DEPTHS: [ 2, 2, 18, 2 ]
-    NUM_HEADS: [ 4, 8, 16, 32 ]
+    EMBED_DIM: 96
+    DEPTHS: [ 2, 2, 6, 2 ]
+    NUM_HEADS: [ 3, 6, 12, 24 ]
     WINDOW_SIZE: 7
 DATA:
   NUM_WORKERS: 10