geometric_classifier v9 | acc=0.9014 | 2,325,174 params | vectorized curvature

geometric_classifier/config.json

@@ -1,6 +1,7 @@
 {
   "model_type": "GeometricShapeClassifier",
-  "version": "v8",
+  "version": "v9",
+  "changes": "conv3d/conv2d \u2192 linear in curvature head, 83x training speedup",
   "grid_size": 5,
   "num_classes": 38,
   "class_names": [
@@ -53,18 +54,11 @@
     "hyperbolic",
     "helical"
   ],
-  "embed_dim": 128,
+  "embed_dim": 64,
   "n_tracers": 5,
-  "capacity_dims": [
-    64,
-    64,
-    64,
-    64
-  ],
-  "curvature_embed_dim": 128,
   "arbiter_latent_dim": 128,
   "arbiter_flow_steps": 4,
-  "total_params": 1852870,
+  "total_params": 2325174,
   "shape_catalog": {
     "point": {
       "dim": 0,

geometric_classifier/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b9d0a0144d9c5746b93fb734ff73df5991cba83e4d25c5cf2258ff17b22c3b1
-size 7433664
+oid sha256:8251db4e5b77cb50df8c0f173e75ffb7cb79cce4e27a20db90e2e2f522354865
+size 9574860

geometric_classifier/training_config.json

@@ -1,10 +1,35 @@
 {
-  "checkpoint_epoch": 79,
-  "best_val_accuracy": 0.90219,
+  "n_samples": 500000,
+  "epochs": 80,
+  "batch_size": 4096,
+  "lr": 0.003,
+  "seed": 42,
+  "optimizer": "AdamW",
+  "weight_decay": 0.0001,
+  "scheduler": "cosine_with_warmup",
+  "warmup_epochs": 5,
+  "amp_dtype": "torch.bfloat16",
+  "loss_weights": {
+    "cls": 1.0,
+    "fill": 0.3,
+    "peak": 0.3,
+    "ovf": 0.05,
+    "div": 0.02,
+    "vol": 0.1,
+    "cm": 0.1,
+    "curved": 0.2,
+    "ctype": 0.2,
+    "arb_cls": 0.8,
+    "arb_traj": 0.2,
+    "arb_conf": 0.1,
+    "flow": 0.5
+  },
+  "best_val_accuracy": 0.90142,
   "learned_capacities": [
-    0.004316786769777536,
-    0.01499153021723032,
-    0.07568630576133728,
-    16.490644454956055
-  ]
+    0.001764793531037867,
+    0.0032583752181380987,
+    18.675533294677734,
+    18.547143936157227
+  ],
+  "total_training_time_seconds": 981.4479973316193
 }