Add EfficientNet-B0 classify config (#1)

* Add EfficientNet-B0 classify config
Y-T-G · Jan 18, 2025 · 2431260 · 2431260
1 parent c8071c4
commit 2431260
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/cfg/classify/efficientnet_b0.yaml b/cfg/classify/efficientnet_b0.yaml
@@ -0,0 +1,18 @@
+# Metadata
+author: Y-T-G
+task: classify
+keywords: [cnn, efficientnet]
+description: EfficientNet-B0 pretrained backbone with Classify head.
+flops: 8.6
+parameters: 5649789
+min_version: 8.3.59
+
+# Model
+nc: 1
+strides: [32]
+backbone:
+  # [from, repeats, module, args]
+  - [-1, 1, TorchVision, [1280, "efficientnet_b0", "DEFAULT", True, 2]]  # (1, 1280, 20, 20) - 0
+
+head:
+  - [-1, 1, Classify, [nc]]