Initial pi0 minimal validation

outputs/overfit/evaluation.json

@@ -0,0 +1,18 @@
+[
+  {
+    "sample_id": "314f59c2-e420-4fcd-9df2-a51d572e0c20_1503899:aug0",
+    "trajectory_mse": 0.1524142026901245
+  },
+  {
+    "sample_id": "314f59c2-e420-4fcd-9df2-a51d572e0c20_1503899:aug1",
+    "trajectory_mse": 0.2658033072948456
+  },
+  {
+    "sample_id": "314f59c2-e420-4fcd-9df2-a51d572e0c20_1503899:aug2",
+    "trajectory_mse": 0.011374758556485176
+  },
+  {
+    "sample_id": "314f59c2-e420-4fcd-9df2-a51d572e0c20_1503899:aug3",
+    "trajectory_mse": 0.2792850136756897
+  }
+]

outputs/overfit/summary.json

@@ -0,0 +1,13 @@
+{
+  "device": "mps",
+  "samples": 4,
+  "parameters": 662786,
+  "initial": {
+    "step": 1,
+    "flow_loss": 2.2617573738098145
+  },
+  "final": {
+    "step": 600,
+    "flow_loss": 0.16418831050395966
+  }
+}

outputs/overfit/training_history.json

@@ -0,0 +1,126 @@
+[
+  {
+    "step": 1,
+    "flow_loss": 2.2617573738098145
+  },
+  {
+    "step": 20,
+    "flow_loss": 1.941188097000122
+  },
+  {
+    "step": 40,
+    "flow_loss": 1.479238510131836
+  },
+  {
+    "step": 60,
+    "flow_loss": 0.7427206635475159
+  },
+  {
+    "step": 80,
+    "flow_loss": 0.6053631901741028
+  },
+  {
+    "step": 100,
+    "flow_loss": 0.270697683095932
+  },
+  {
+    "step": 120,
+    "flow_loss": 0.11459995806217194
+  },
+  {
+    "step": 140,
+    "flow_loss": 0.4224633276462555
+  },
+  {
+    "step": 160,
+    "flow_loss": 0.14150527119636536
+  },
+  {
+    "step": 180,
+    "flow_loss": 0.15842989087104797
+  },
+  {
+    "step": 200,
+    "flow_loss": 0.34323638677597046
+  },
+  {
+    "step": 220,
+    "flow_loss": 0.1762494295835495
+  },
+  {
+    "step": 240,
+    "flow_loss": 0.3678283989429474
+  },
+  {
+    "step": 260,
+    "flow_loss": 0.10764116048812866
+  },
+  {
+    "step": 280,
+    "flow_loss": 0.17170406877994537
+  },
+  {
+    "step": 300,
+    "flow_loss": 0.20748832821846008
+  },
+  {
+    "step": 320,
+    "flow_loss": 0.12005910277366638
+  },
+  {
+    "step": 340,
+    "flow_loss": 0.1855630874633789
+  },
+  {
+    "step": 360,
+    "flow_loss": 0.16064667701721191
+  },
+  {
+    "step": 380,
+    "flow_loss": 0.11575131863355637
+  },
+  {
+    "step": 400,
+    "flow_loss": 0.1507677137851715
+  },
+  {
+    "step": 420,
+    "flow_loss": 0.23872917890548706
+  },
+  {
+    "step": 440,
+    "flow_loss": 0.25908488035202026
+  },
+  {
+    "step": 460,
+    "flow_loss": 0.15245740115642548
+  },
+  {
+    "step": 480,
+    "flow_loss": 0.3529265224933624
+  },
+  {
+    "step": 500,
+    "flow_loss": 0.22890940308570862
+  },
+  {
+    "step": 520,
+    "flow_loss": 0.18230678141117096
+  },
+  {
+    "step": 540,
+    "flow_loss": 0.14777007699012756
+  },
+  {
+    "step": 560,
+    "flow_loss": 0.06152040883898735
+  },
+  {
+    "step": 580,
+    "flow_loss": 0.18225648999214172
+  },
+  {
+    "step": 600,
+    "flow_loss": 0.16418831050395966
+  }
+]