Upload 3 files

config.json

@@ -5,18 +5,10 @@
   "data": {
     "columns": null,
     "dtype": "float32",
-    "inputs": [
-      0,
-      1,
-      2
-    ],
+    "inputs": [],
     "memmap": true,
-    "path": "develop_data/isochrones.npy",
-    "targets": [
-      3,
-      4,
-      5
-    ]
+    "path": "",
+    "targets": []
   },
   "hub": {
     "repo_id": null,
@@ -24,9 +16,17 @@
   },
   "io": {
     "x_dim": 3,
-    "x_names": null,
+    "x_names": [
+      "age",
+      "eep",
+      "feh"
+    ],
     "y_dim": 3,
-    "y_names": null
+    "y_names": [
+      "G_mag",
+      "BP_mag",
+      "RP_mag"
+    ]
   },
   "model": {
     "name": "mlp",
@@ -34,43 +34,52 @@
       "activation": "gelu",
       "dtype": "float32",
       "hidden_sizes": [
-        32,
-        32
+        64,
+        64,
+        64
       ]
     }
   },
   "optim": {
     "b1": 0.9,
     "b2": 0.999,
+    "decay_steps": 0,
     "eps": 1e-08,
-    "lr": 0.0003,
-    "name": "adamw",
-    "weight_decay": 0.0001
-  },
-  "preprocess": {
-    "eps": 1e-08,
-    "x_scaler": "standard",
-    "y_scaler": "standard"
+    "lr": 0.003,
+    "name": "soap",
+    "precondition_1d": false,
+    "precondition_frequency": 10,
+    "schedule": "cosine",
+    "warmup_steps": 10000,
+    "weight_decay": 1e-05
   },
   "schema_version": 1,
   "seed": 0,
   "task": {
     "name": "regression",
     "params": {
-      "loss": "mse"
+      "loss": "mse",
+      "metric_axes": {
+        "global": "all",
+        "per_dim": []
+      },
+      "metrics": [
+        "mse",
+        "mae"
+      ]
     }
   },
   "training": {
     "batch_size": 2048,
-    "checkpoint_every_steps": 200,
-    "eval_every_steps": 200,
-    "log_every_steps": 25,
+    "checkpoint_every_steps": 150000,
+    "eval_every_steps": 5000,
+    "log_every_steps": 1000,
     "max_checkpoints": 3,
-    "num_epochs": 1,
+    "num_steps": 100000,
     "shuffle": true,
     "shuffle_seed": 0,
     "steps_per_epoch": null,
     "val_fraction": 0.1,
-    "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/runs/iso_mlp"
+    "workdir": "/Users/tr/repos/astro-emulators-toolkit/large_local_examples/isochrones/run2"
   }
 }

metadata.json

@@ -1,4 +1,23 @@
 {
+  "fit_method": "gradient",
   "schema_version": 1,
-  "tool": "astro_emulators_toolkit"
+  "tool": "astro_emulators_toolkit",
+  "x_schema": {
+    "dim": 3,
+    "names": [
+      "age",
+      "eep",
+      "feh"
+    ],
+    "representation": "model-space inputs emitted directly by the dataset"
+  },
+  "y_schema": {
+    "dim": 3,
+    "names": [
+      "G_mag",
+      "BP_mag",
+      "RP_mag"
+    ],
+    "representation": "model-space targets emitted directly by the dataset"
+  }
 }

weights.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63e1b1b370cfea3fdc99314689db09cfb38bb282075d42ac0ef56436799621d9
-size 5620
+oid sha256:7916af20b55930166ba2f40866892ecf13cec5c9354d827ac44055d6797bf8e8
+size 35876