open-reason-small / train_metrics.json
theworker02's picture
Upload Open Reason small CPU causal LM (not 1B)
bf0b9a3 verified
Raw
History Blame Contribute Delete
643 Bytes
{
"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."
}