pi0-minimal-validation/outputs/overfit/summary.json

{
  "device": "mps",
  "samples": 4,
  "parameters": 662786,
  "initial": {
    "step": 1,
    "flow_loss": 2.2617573738098145
  },
  "final": {
    "step": 600,
    "flow_loss": 0.16418831050395966
  }
}