| { |
| "history": [ |
| { |
| "epoch": 0, |
| "phase": "ce", |
| "global_step": 2251, |
| "train_L_dec": 0.3132325249438689, |
| "train_L_cal": 0.022969920731528693, |
| "train_acc": 0.8643241892492226, |
| "val_acc": 0.9746167518329261, |
| "val_brier": 0.004308474582756061, |
| "val_ece": 0.00984656126080116, |
| "seconds": 1176.4715111255646, |
| "stability_spike_warnings_total": 2 |
| }, |
| { |
| "epoch": 1, |
| "phase": "ce", |
| "global_step": 4502, |
| "train_L_dec": 0.03158559516649023, |
| "train_L_cal": 0.0016624860179579265, |
| "train_acc": 0.9905389271434918, |
| "val_acc": 0.9796711841812931, |
| "val_brier": 0.0036710468228806767, |
| "val_ece": 0.010786583753486019, |
| "seconds": 1174.2428555488586, |
| "stability_spike_warnings_total": 5 |
| }, |
| { |
| "epoch": 2, |
| "phase": "pcgrad", |
| "global_step": 6753, |
| "train_L_dec": 0.021729152236390645, |
| "train_L_cal": 0.0011230140197688602, |
| "train_acc": 0.9936347734340293, |
| "val_acc": 0.9828371473005999, |
| "val_brier": 0.0029678672562138916, |
| "val_ece": 0.007705523323957138, |
| "seconds": 1947.3289678096771, |
| "stability_spike_warnings_total": 5 |
| }, |
| { |
| "epoch": 3, |
| "phase": "pcgrad", |
| "global_step": 9004, |
| "train_L_dec": 0.01443692476023041, |
| "train_L_cal": 0.0007580307265767108, |
| "train_acc": 0.995620002221235, |
| "val_acc": 0.9836702954898912, |
| "val_brier": 0.002991214513442398, |
| "val_ece": 0.008586534102594447, |
| "seconds": 1941.522153377533, |
| "stability_spike_warnings_total": 11 |
| }, |
| { |
| "epoch": 4, |
| "phase": "pcgrad", |
| "global_step": 11255, |
| "train_L_dec": 0.012373026743064744, |
| "train_L_cal": 0.00065262238645809, |
| "train_acc": 0.9961614282541092, |
| "val_acc": 0.9843368140413241, |
| "val_brier": 0.002747937278662336, |
| "val_ece": 0.009184632024925778, |
| "seconds": 1943.6136224269867, |
| "stability_spike_warnings_total": 18 |
| }, |
| { |
| "epoch": 5, |
| "phase": "pcgrad", |
| "global_step": 13506, |
| "train_L_dec": 0.010130272916737846, |
| "train_L_cal": 0.0004837604737681557, |
| "train_acc": 0.9971001450351374, |
| "val_acc": 0.9843368140413241, |
| "val_brier": 0.0026608713938844693, |
| "val_ece": 0.009472633415071256, |
| "seconds": 1943.7929818630219, |
| "stability_spike_warnings_total": 48 |
| }, |
| { |
| "epoch": 6, |
| "phase": "pcgrad", |
| "global_step": 15757, |
| "train_L_dec": 0.008186050639382882, |
| "train_L_cal": 0.0003799829597298574, |
| "train_acc": 0.9977093513993781, |
| "val_acc": 0.9847811597422795, |
| "val_brier": 0.0027190133560318773, |
| "val_ece": 0.009519949778864792, |
| "seconds": 1945.6040878295898, |
| "stability_spike_warnings_total": 83 |
| }, |
| { |
| "epoch": 7, |
| "phase": "pcgrad", |
| "global_step": 18008, |
| "train_L_dec": 0.009192034906286452, |
| "train_L_cal": 0.0004240516285463041, |
| "train_acc": 0.9975358174144825, |
| "val_acc": 0.9848922461675184, |
| "val_brier": 0.0026594605793753444, |
| "val_ece": 0.009249331458982588, |
| "seconds": 1939.7083559036255, |
| "stability_spike_warnings_total": 128 |
| } |
| ], |
| "best_acc": { |
| "acc": 0.9848922461675184, |
| "brier": 0.0026594605793753444, |
| "epoch": 7 |
| }, |
| "best_brier": { |
| "brier": 0.0026594605793753444, |
| "acc": 0.9848922461675184, |
| "epoch": 7 |
| }, |
| "split_sizes": { |
| "train": 144017, |
| "val": 18004, |
| "holdout": 18002 |
| }, |
| "pretrained_audit": { |
| "wrapper_class": "Lfm2BidirectionalForMaskedLM", |
| "encoder_class": "Lfm2BidirectionalModel", |
| "missing_keys": [], |
| "unexpected_keys": [], |
| "parameter_count": 354483968, |
| "fingerprints": { |
| "embed_tokens.weight": "b9d03e9055d7428a6467f81860d5ebb58286174da60f58364ecd66fdcfa58dd3", |
| "layers.0.feed_forward.w1.weight": "86a63d2c73d6334c58dba4e68282f97e70e9dc1342b3b06bb83b8451b86b4f69" |
| } |
| }, |
| "total_params": 357631745, |
| "trainable_params": 357631745, |
| "stability_spike_warnings_total": 128 |
| } |
