Add Swin-T classify config (#2)

Y-T-G · Jan 18, 2025 · c044ebc · c044ebc
1 parent 21cc228
commit c044ebc
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/cfg/classify/swin_t.yaml b/cfg/classify/swin_t.yaml
@@ -0,0 +1,18 @@
+# Metadata
+author: Y-T-G
+task: classify
+keywords: [transformer, swin]
+description: Swin-T pretrained backbone with Classify head.
+flops: 49.4
+parameters: 28506235
+min_version: 8.3.59
+
+# Model
+nc: 1
+strides: [32]
+backbone:
+  # [from, repeats, module, args]
+  - [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 3]]  # (1, 768, 20, 20) - 0
+
+head:
+  - [-1, 1, Classify, [nc]]