From c044ebc55b5418737bd72e6a5a642b874ad18bf9 Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Sun, 19 Jan 2025 00:05:55 +0800
Subject: [PATCH] Add Swin-T classify config (#2)

---
 cfg/classify/swin_t.yaml | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 cfg/classify/swin_t.yaml

diff --git a/cfg/classify/swin_t.yaml b/cfg/classify/swin_t.yaml
new file mode 100644
index 0000000..c0f62bd
--- /dev/null
+++ 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]]
\ No newline at end of file