1-parameter-classifier / stage_4 /training_log.json
phanerozoic's picture
Stage 4: specialist student (3.27M params, F1 0.710 vs 0.894 baseline)
864ba61 verified
{
"epochs": [
{
"epoch": 1,
"loss": 2.2497359863064306,
"F1": 0.7068741917610168,
"precision": 0.5471887588500977,
"recall": 0.9981684684753418,
"threshold": 27.844703674316406
},
{
"epoch": 2,
"loss": 2.006192959226011,
"F1": 0.7077131867408752,
"precision": 0.5611587762832642,
"recall": 0.9578754305839539,
"threshold": 29.012691497802734
},
{
"epoch": 3,
"loss": 2.0050793833516254,
"F1": 0.7173912525177002,
"precision": 0.5701943635940552,
"recall": 0.9670329689979553,
"threshold": 26.3531494140625
},
{
"epoch": 4,
"loss": 2.008850994551679,
"F1": 0.7111111283302307,
"precision": 0.5528455376625061,
"recall": 0.9963369965553284,
"threshold": 26.654373168945312
},
{
"epoch": 5,
"loss": 1.9956948308571767,
"F1": 0.7116155028343201,
"precision": 0.5598739385604858,
"recall": 0.976190447807312,
"threshold": 26.531875610351562
},
{
"epoch": 6,
"loss": 2.001336439533982,
"F1": 0.7114093899726868,
"precision": 0.5614407062530518,
"recall": 0.970695972442627,
"threshold": 25.896560668945312
},
{
"epoch": 7,
"loss": 1.9895635365096358,
"F1": 0.7127659916877747,
"precision": 0.5594989657402039,
"recall": 0.9816849827766418,
"threshold": 26.533084869384766
},
{
"epoch": 8,
"loss": 1.9920095616759776,
"F1": 0.7133058905601501,
"precision": 0.5701754093170166,
"recall": 0.9523809552192688,
"threshold": 26.670108795166016
},
{
"epoch": 9,
"loss": 1.9944281049740566,
"F1": 0.7095890045166016,
"precision": 0.5667396187782288,
"recall": 0.9487179517745972,
"threshold": 26.39623260498047
},
{
"epoch": 10,
"loss": 1.9913188398937247,
"F1": 0.7101648449897766,
"precision": 0.5681318640708923,
"recall": 0.946886420249939,
"threshold": 26.328819274902344
}
],
"student_params": 3267304
}