ThePyProgrammer's picture
Upload trainer_state.json
d036598
{
"best_metric": 0.6546083688735962,
"best_model_checkpoint": "models_gitignored/distilbert-base-uncased-finetuned-sentence-classification/checkpoint-12626",
"epoch": 4.0,
"global_step": 50504,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"learning_rate": 1.984159670521147e-05,
"loss": 1.177,
"step": 500
},
{
"epoch": 0.08,
"learning_rate": 1.968319341042294e-05,
"loss": 0.9186,
"step": 1000
},
{
"epoch": 0.12,
"learning_rate": 1.9524790115634406e-05,
"loss": 0.8261,
"step": 1500
},
{
"epoch": 0.16,
"learning_rate": 1.9366386820845877e-05,
"loss": 0.8177,
"step": 2000
},
{
"epoch": 0.2,
"learning_rate": 1.9207983526057344e-05,
"loss": 0.7788,
"step": 2500
},
{
"epoch": 0.24,
"learning_rate": 1.904958023126881e-05,
"loss": 0.7519,
"step": 3000
},
{
"epoch": 0.28,
"learning_rate": 1.889117693648028e-05,
"loss": 0.7243,
"step": 3500
},
{
"epoch": 0.32,
"learning_rate": 1.8732773641691748e-05,
"loss": 0.7431,
"step": 4000
},
{
"epoch": 0.36,
"learning_rate": 1.8574370346903215e-05,
"loss": 0.7297,
"step": 4500
},
{
"epoch": 0.4,
"learning_rate": 1.8415967052114686e-05,
"loss": 0.7061,
"step": 5000
},
{
"epoch": 0.44,
"learning_rate": 1.8257563757326153e-05,
"loss": 0.698,
"step": 5500
},
{
"epoch": 0.48,
"learning_rate": 1.8099160462537623e-05,
"loss": 0.7074,
"step": 6000
},
{
"epoch": 0.51,
"learning_rate": 1.794075716774909e-05,
"loss": 0.6975,
"step": 6500
},
{
"epoch": 0.55,
"learning_rate": 1.7782353872960557e-05,
"loss": 0.6943,
"step": 7000
},
{
"epoch": 0.59,
"learning_rate": 1.7623950578172028e-05,
"loss": 0.7045,
"step": 7500
},
{
"epoch": 0.63,
"learning_rate": 1.7465547283383495e-05,
"loss": 0.6764,
"step": 8000
},
{
"epoch": 0.67,
"learning_rate": 1.730714398859496e-05,
"loss": 0.6722,
"step": 8500
},
{
"epoch": 0.71,
"learning_rate": 1.7148740693806432e-05,
"loss": 0.6884,
"step": 9000
},
{
"epoch": 0.75,
"learning_rate": 1.69903373990179e-05,
"loss": 0.6903,
"step": 9500
},
{
"epoch": 0.79,
"learning_rate": 1.683193410422937e-05,
"loss": 0.6854,
"step": 10000
},
{
"epoch": 0.83,
"learning_rate": 1.667353080944084e-05,
"loss": 0.6861,
"step": 10500
},
{
"epoch": 0.87,
"learning_rate": 1.6515127514652307e-05,
"loss": 0.6894,
"step": 11000
},
{
"epoch": 0.91,
"learning_rate": 1.6356724219863774e-05,
"loss": 0.6754,
"step": 11500
},
{
"epoch": 0.95,
"learning_rate": 1.6198320925075244e-05,
"loss": 0.6918,
"step": 12000
},
{
"epoch": 0.99,
"learning_rate": 1.603991763028671e-05,
"loss": 0.6746,
"step": 12500
},
{
"epoch": 1.0,
"eval_accuracy": 0.778547431856752,
"eval_f1": 0.7757136063142371,
"eval_kappa": 0.7014269317598121,
"eval_loss": 0.6546083688735962,
"eval_precision": 0.7762465189891471,
"eval_recall": 0.778547431856752,
"eval_runtime": 150.616,
"eval_samples_per_second": 191.699,
"eval_steps_per_second": 23.968,
"step": 12626
},
{
"epoch": 1.03,
"learning_rate": 1.5881514335498182e-05,
"loss": 0.5694,
"step": 13000
},
{
"epoch": 1.07,
"learning_rate": 1.572311104070965e-05,
"loss": 0.5509,
"step": 13500
},
{
"epoch": 1.11,
"learning_rate": 1.5564707745921116e-05,
"loss": 0.5466,
"step": 14000
},
{
"epoch": 1.15,
"learning_rate": 1.5406304451132586e-05,
"loss": 0.5544,
"step": 14500
},
{
"epoch": 1.19,
"learning_rate": 1.5247901156344053e-05,
"loss": 0.566,
"step": 15000
},
{
"epoch": 1.23,
"learning_rate": 1.5089497861555522e-05,
"loss": 0.5773,
"step": 15500
},
{
"epoch": 1.27,
"learning_rate": 1.4931094566766991e-05,
"loss": 0.6056,
"step": 16000
},
{
"epoch": 1.31,
"learning_rate": 1.4772691271978458e-05,
"loss": 0.5532,
"step": 16500
},
{
"epoch": 1.35,
"learning_rate": 1.4614287977189927e-05,
"loss": 0.5764,
"step": 17000
},
{
"epoch": 1.39,
"learning_rate": 1.4455884682401395e-05,
"loss": 0.5499,
"step": 17500
},
{
"epoch": 1.43,
"learning_rate": 1.4297481387612864e-05,
"loss": 0.5695,
"step": 18000
},
{
"epoch": 1.47,
"learning_rate": 1.4139078092824333e-05,
"loss": 0.5823,
"step": 18500
},
{
"epoch": 1.5,
"learning_rate": 1.39806747980358e-05,
"loss": 0.5329,
"step": 19000
},
{
"epoch": 1.54,
"learning_rate": 1.3822271503247269e-05,
"loss": 0.5661,
"step": 19500
},
{
"epoch": 1.58,
"learning_rate": 1.3663868208458737e-05,
"loss": 0.5851,
"step": 20000
},
{
"epoch": 1.62,
"learning_rate": 1.3505464913670206e-05,
"loss": 0.5859,
"step": 20500
},
{
"epoch": 1.66,
"learning_rate": 1.3347061618881673e-05,
"loss": 0.5569,
"step": 21000
},
{
"epoch": 1.7,
"learning_rate": 1.3188658324093142e-05,
"loss": 0.5473,
"step": 21500
},
{
"epoch": 1.74,
"learning_rate": 1.303025502930461e-05,
"loss": 0.5635,
"step": 22000
},
{
"epoch": 1.78,
"learning_rate": 1.287185173451608e-05,
"loss": 0.5848,
"step": 22500
},
{
"epoch": 1.82,
"learning_rate": 1.2713448439727546e-05,
"loss": 0.5885,
"step": 23000
},
{
"epoch": 1.86,
"learning_rate": 1.2555045144939015e-05,
"loss": 0.5502,
"step": 23500
},
{
"epoch": 1.9,
"learning_rate": 1.2396641850150484e-05,
"loss": 0.5999,
"step": 24000
},
{
"epoch": 1.94,
"learning_rate": 1.2238238555361953e-05,
"loss": 0.5639,
"step": 24500
},
{
"epoch": 1.98,
"learning_rate": 1.207983526057342e-05,
"loss": 0.5664,
"step": 25000
},
{
"epoch": 2.0,
"eval_accuracy": 0.7842274789595817,
"eval_f1": 0.7824161012799614,
"eval_kappa": 0.7102691347246439,
"eval_loss": 0.7084277868270874,
"eval_precision": 0.7870631351341026,
"eval_recall": 0.7842274789595817,
"eval_runtime": 152.0087,
"eval_samples_per_second": 189.943,
"eval_steps_per_second": 23.749,
"step": 25252
},
{
"epoch": 2.02,
"learning_rate": 1.1921431965784888e-05,
"loss": 0.4751,
"step": 25500
},
{
"epoch": 2.06,
"learning_rate": 1.1763028670996357e-05,
"loss": 0.4208,
"step": 26000
},
{
"epoch": 2.1,
"learning_rate": 1.1604625376207826e-05,
"loss": 0.4563,
"step": 26500
},
{
"epoch": 2.14,
"learning_rate": 1.1446222081419293e-05,
"loss": 0.4513,
"step": 27000
},
{
"epoch": 2.18,
"learning_rate": 1.1287818786630762e-05,
"loss": 0.4385,
"step": 27500
},
{
"epoch": 2.22,
"learning_rate": 1.112941549184223e-05,
"loss": 0.4493,
"step": 28000
},
{
"epoch": 2.26,
"learning_rate": 1.0971012197053699e-05,
"loss": 0.4533,
"step": 28500
},
{
"epoch": 2.3,
"learning_rate": 1.0812608902265168e-05,
"loss": 0.4365,
"step": 29000
},
{
"epoch": 2.34,
"learning_rate": 1.0654205607476635e-05,
"loss": 0.4205,
"step": 29500
},
{
"epoch": 2.38,
"learning_rate": 1.0495802312688104e-05,
"loss": 0.4538,
"step": 30000
},
{
"epoch": 2.42,
"learning_rate": 1.0337399017899574e-05,
"loss": 0.4653,
"step": 30500
},
{
"epoch": 2.46,
"learning_rate": 1.0178995723111043e-05,
"loss": 0.436,
"step": 31000
},
{
"epoch": 2.49,
"learning_rate": 1.0020592428322511e-05,
"loss": 0.4249,
"step": 31500
},
{
"epoch": 2.53,
"learning_rate": 9.862189133533979e-06,
"loss": 0.4441,
"step": 32000
},
{
"epoch": 2.57,
"learning_rate": 9.703785838745447e-06,
"loss": 0.4544,
"step": 32500
},
{
"epoch": 2.61,
"learning_rate": 9.545382543956914e-06,
"loss": 0.446,
"step": 33000
},
{
"epoch": 2.65,
"learning_rate": 9.386979249168383e-06,
"loss": 0.4457,
"step": 33500
},
{
"epoch": 2.69,
"learning_rate": 9.228575954379852e-06,
"loss": 0.4472,
"step": 34000
},
{
"epoch": 2.73,
"learning_rate": 9.07017265959132e-06,
"loss": 0.4728,
"step": 34500
},
{
"epoch": 2.77,
"learning_rate": 8.911769364802788e-06,
"loss": 0.4455,
"step": 35000
},
{
"epoch": 2.81,
"learning_rate": 8.753366070014258e-06,
"loss": 0.4622,
"step": 35500
},
{
"epoch": 2.85,
"learning_rate": 8.594962775225727e-06,
"loss": 0.4566,
"step": 36000
},
{
"epoch": 2.89,
"learning_rate": 8.436559480437194e-06,
"loss": 0.4185,
"step": 36500
},
{
"epoch": 2.93,
"learning_rate": 8.278156185648662e-06,
"loss": 0.4351,
"step": 37000
},
{
"epoch": 2.97,
"learning_rate": 8.119752890860131e-06,
"loss": 0.4566,
"step": 37500
},
{
"epoch": 3.0,
"eval_accuracy": 0.7792747549613827,
"eval_f1": 0.7780530572128844,
"eval_kappa": 0.7039084509242337,
"eval_loss": 0.7550320029258728,
"eval_precision": 0.7813618395098462,
"eval_recall": 0.7792747549613827,
"eval_runtime": 151.5948,
"eval_samples_per_second": 190.462,
"eval_steps_per_second": 23.813,
"step": 37878
},
{
"epoch": 3.01,
"learning_rate": 7.9613495960716e-06,
"loss": 0.4023,
"step": 38000
},
{
"epoch": 3.05,
"learning_rate": 7.802946301283067e-06,
"loss": 0.3119,
"step": 38500
},
{
"epoch": 3.09,
"learning_rate": 7.644543006494536e-06,
"loss": 0.3205,
"step": 39000
},
{
"epoch": 3.13,
"learning_rate": 7.4861397117060044e-06,
"loss": 0.35,
"step": 39500
},
{
"epoch": 3.17,
"learning_rate": 7.327736416917472e-06,
"loss": 0.3333,
"step": 40000
},
{
"epoch": 3.21,
"learning_rate": 7.169333122128941e-06,
"loss": 0.3563,
"step": 40500
},
{
"epoch": 3.25,
"learning_rate": 7.010929827340409e-06,
"loss": 0.3388,
"step": 41000
},
{
"epoch": 3.29,
"learning_rate": 6.852526532551878e-06,
"loss": 0.3422,
"step": 41500
},
{
"epoch": 3.33,
"learning_rate": 6.6941232377633456e-06,
"loss": 0.3336,
"step": 42000
},
{
"epoch": 3.37,
"learning_rate": 6.535719942974814e-06,
"loss": 0.3433,
"step": 42500
},
{
"epoch": 3.41,
"learning_rate": 6.377316648186282e-06,
"loss": 0.3574,
"step": 43000
},
{
"epoch": 3.45,
"learning_rate": 6.218913353397751e-06,
"loss": 0.3105,
"step": 43500
},
{
"epoch": 3.48,
"learning_rate": 6.060510058609219e-06,
"loss": 0.3545,
"step": 44000
},
{
"epoch": 3.52,
"learning_rate": 5.9021067638206875e-06,
"loss": 0.3469,
"step": 44500
},
{
"epoch": 3.56,
"learning_rate": 5.7437034690321554e-06,
"loss": 0.35,
"step": 45000
},
{
"epoch": 3.6,
"learning_rate": 5.585300174243625e-06,
"loss": 0.3425,
"step": 45500
},
{
"epoch": 3.64,
"learning_rate": 5.426896879455094e-06,
"loss": 0.3203,
"step": 46000
},
{
"epoch": 3.68,
"learning_rate": 5.268493584666562e-06,
"loss": 0.3549,
"step": 46500
},
{
"epoch": 3.72,
"learning_rate": 5.11009028987803e-06,
"loss": 0.3326,
"step": 47000
},
{
"epoch": 3.76,
"learning_rate": 4.951686995089498e-06,
"loss": 0.3328,
"step": 47500
},
{
"epoch": 3.8,
"learning_rate": 4.793283700300967e-06,
"loss": 0.3376,
"step": 48000
},
{
"epoch": 3.84,
"learning_rate": 4.634880405512435e-06,
"loss": 0.3285,
"step": 48500
},
{
"epoch": 3.88,
"learning_rate": 4.476477110723904e-06,
"loss": 0.361,
"step": 49000
},
{
"epoch": 3.92,
"learning_rate": 4.3180738159353715e-06,
"loss": 0.3487,
"step": 49500
},
{
"epoch": 3.96,
"learning_rate": 4.15967052114684e-06,
"loss": 0.3511,
"step": 50000
},
{
"epoch": 4.0,
"learning_rate": 4.001267226358309e-06,
"loss": 0.3528,
"step": 50500
},
{
"epoch": 4.0,
"eval_accuracy": 0.7778201087521214,
"eval_f1": 0.7760540283760836,
"eval_kappa": 0.7007853270915267,
"eval_loss": 0.9983726739883423,
"eval_precision": 0.7770574865770817,
"eval_recall": 0.7778201087521214,
"eval_runtime": 151.818,
"eval_samples_per_second": 190.182,
"eval_steps_per_second": 23.778,
"step": 50504
}
],
"max_steps": 63130,
"num_train_epochs": 5,
"total_flos": 5.330919857118713e+16,
"trial_name": null,
"trial_params": null
}