update

gabrieldernbach · Mar 10, 2024 · 6c4ef5a · 6c4ef5a
1 parent 2136254
commit 6c4ef5a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/classification/configs/vssm1/vssm_tiny_224_0312.yaml b/classification/configs/vssm1/vssm_tiny_224_0312.yaml
@@ -0,0 +1,16 @@
+MODEL:
+  TYPE: vssm
+  NAME: vssm1_tiny_0312
+  DROP_PATH_RATE: 0.2
+  VSSM:
+    EMBED_DIM: 96
+    DEPTHS: [ 2, 2, 5, 2 ]
+    SSM_D_STATE: 1
+    SSM_DT_RANK: "auto"
+    SSM_RATIO: 2.0
+    SSM_CONV: 3
+    SSM_CONV_BIAS: false
+    SSM_FORWARDTYPE: "xv3noz"
+    MLP_RATIO: 4.0
+    DOWNSAMPLE: "v3"
+    PATCHEMBED: "v2"