{
  "smoke": false,
  "cuda": false,
  "backend": "cpu-host",
  "architecture": "gpt2-scratch",
  "param_count": 1334016,
  "n_layer": 4,
  "n_embd": 128,
  "n_head": 4,
  "steps": 200,
  "rows": 2348,
  "final_loss": 6.095415115356445,
  "losses_tail": [
    5.3571391105651855,
    5.712825298309326,
    5.500826835632324,
    5.698274612426758,
    5.465966701507568,
    5.777740001678467,
    5.92188024520874,
    5.687778472900391,
    5.9522929191589355,
    6.095415115356445
  ],
  "hub_id_if_uploaded": "theworker02/open-reason-small",
  "note": "Small CPU causal LM. Not open-reason-1b. Not AMD GPU."
}