Archive Kurome classifier: AnatomyFlaws-v2.9_centercrop_attn_ce.config.json

kurome/classifiers/anatomy-flaws/AnatomyFlaws-v2.9_centercrop_attn_ce.config.json

@@ -0,0 +1,42 @@
+{
+  "model": {
+    "base": "AnatomyFlaws",
+    "rev": "v2.9_centercrop_attn_ce",
+    "arch": "class",
+    "embed_ver": "siglip2_so400m_patch16_512_CenterCrop",
+    "base_vision_model": "google/siglip2-so400m-patch16-512",
+    "num_attn_heads": 8,
+    "attn_dropout": 0.1
+  },
+  "train": {
+    "lr": 0.0001,
+    "steps": 50000,
+    "batch": 32,
+    "optimizer": "ADOPTAOScheduleFree",
+    "loss_function": "crossentropy",
+    "precision": "fp32",
+    "val_split_count": 60,
+    "betas": [
+      0.9,
+      0.999
+    ],
+    "eps": "1e-6",
+    "weight_decay": "1e-3",
+    "cosine": true,
+    "warmup_steps": 0,
+    "r_sf": 0.0,
+    "wlpow_sf": 2.0,
+    "state_precision": "parameter",
+    "adaptive_clip": 1.0,
+    "adaptive_clip_eps": "1e-3"
+  },
+  "labels": {
+    "0": "Bad Anatomy",
+    "1": "Good Anatomy"
+  },
+  "model_params": {
+    "features": 1152,
+    "hidden": 1280,
+    "outputs": 2
+  }
+}