gpt2-810M-2.2B-USPTOAndPubMedAbs / trainer_state.json
Allison Casasola
files
f3766ad
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 68000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 4.9632352941176476e-05,
"loss": 6.14,
"step": 500
},
{
"epoch": 0.01,
"learning_rate": 4.9264705882352944e-05,
"loss": 5.1656,
"step": 1000
},
{
"epoch": 0.02,
"learning_rate": 4.889705882352941e-05,
"loss": 4.7511,
"step": 1500
},
{
"epoch": 0.03,
"learning_rate": 4.8529411764705885e-05,
"loss": 4.436,
"step": 2000
},
{
"epoch": 0.04,
"learning_rate": 4.816176470588236e-05,
"loss": 4.2339,
"step": 2500
},
{
"epoch": 0.04,
"learning_rate": 4.7794117647058826e-05,
"loss": 4.2329,
"step": 3000
},
{
"epoch": 0.05,
"learning_rate": 4.742647058823529e-05,
"loss": 3.9944,
"step": 3500
},
{
"epoch": 0.06,
"learning_rate": 4.705882352941177e-05,
"loss": 3.8267,
"step": 4000
},
{
"epoch": 0.07,
"learning_rate": 4.669117647058824e-05,
"loss": 3.6853,
"step": 4500
},
{
"epoch": 0.07,
"learning_rate": 4.632352941176471e-05,
"loss": 3.5696,
"step": 5000
},
{
"epoch": 0.08,
"learning_rate": 4.5955882352941176e-05,
"loss": 3.5005,
"step": 5500
},
{
"epoch": 0.09,
"learning_rate": 4.558823529411765e-05,
"loss": 3.4346,
"step": 6000
},
{
"epoch": 0.1,
"learning_rate": 4.522058823529412e-05,
"loss": 3.3698,
"step": 6500
},
{
"epoch": 0.1,
"learning_rate": 4.485294117647059e-05,
"loss": 3.3167,
"step": 7000
},
{
"epoch": 0.11,
"learning_rate": 4.448529411764706e-05,
"loss": 3.2831,
"step": 7500
},
{
"epoch": 0.12,
"learning_rate": 4.411764705882353e-05,
"loss": 3.2386,
"step": 8000
},
{
"epoch": 0.12,
"learning_rate": 4.375e-05,
"loss": 3.2127,
"step": 8500
},
{
"epoch": 0.13,
"learning_rate": 4.3382352941176474e-05,
"loss": 3.1792,
"step": 9000
},
{
"epoch": 0.14,
"learning_rate": 4.301470588235295e-05,
"loss": 3.2568,
"step": 9500
},
{
"epoch": 0.15,
"learning_rate": 4.2647058823529415e-05,
"loss": 3.1827,
"step": 10000
},
{
"epoch": 0.15,
"learning_rate": 4.227941176470588e-05,
"loss": 3.1613,
"step": 10500
},
{
"epoch": 0.16,
"learning_rate": 4.1911764705882356e-05,
"loss": 3.1282,
"step": 11000
},
{
"epoch": 0.17,
"learning_rate": 4.154411764705883e-05,
"loss": 3.1111,
"step": 11500
},
{
"epoch": 0.18,
"learning_rate": 4.11764705882353e-05,
"loss": 3.0499,
"step": 12000
},
{
"epoch": 0.18,
"learning_rate": 4.0808823529411765e-05,
"loss": 3.0571,
"step": 12500
},
{
"epoch": 0.19,
"learning_rate": 4.044117647058824e-05,
"loss": 3.0254,
"step": 13000
},
{
"epoch": 0.2,
"learning_rate": 4.007352941176471e-05,
"loss": 2.9945,
"step": 13500
},
{
"epoch": 0.21,
"learning_rate": 3.970588235294117e-05,
"loss": 3.0066,
"step": 14000
},
{
"epoch": 0.21,
"learning_rate": 3.933823529411765e-05,
"loss": 2.9107,
"step": 14500
},
{
"epoch": 0.22,
"learning_rate": 3.897058823529412e-05,
"loss": 2.9672,
"step": 15000
},
{
"epoch": 0.23,
"learning_rate": 3.8602941176470595e-05,
"loss": 3.0046,
"step": 15500
},
{
"epoch": 0.24,
"learning_rate": 3.8235294117647055e-05,
"loss": 3.0522,
"step": 16000
},
{
"epoch": 0.24,
"learning_rate": 3.786764705882353e-05,
"loss": 3.0229,
"step": 16500
},
{
"epoch": 0.25,
"learning_rate": 3.7500000000000003e-05,
"loss": 3.0024,
"step": 17000
},
{
"epoch": 0.26,
"learning_rate": 3.713235294117647e-05,
"loss": 2.9769,
"step": 17500
},
{
"epoch": 0.26,
"learning_rate": 3.6764705882352945e-05,
"loss": 2.9839,
"step": 18000
},
{
"epoch": 0.27,
"learning_rate": 3.639705882352941e-05,
"loss": 2.9466,
"step": 18500
},
{
"epoch": 0.28,
"learning_rate": 3.6029411764705886e-05,
"loss": 2.965,
"step": 19000
},
{
"epoch": 0.29,
"learning_rate": 3.566176470588235e-05,
"loss": 2.9404,
"step": 19500
},
{
"epoch": 0.29,
"learning_rate": 3.529411764705883e-05,
"loss": 2.8946,
"step": 20000
},
{
"epoch": 0.3,
"learning_rate": 3.4926470588235294e-05,
"loss": 2.8613,
"step": 20500
},
{
"epoch": 0.31,
"learning_rate": 3.455882352941177e-05,
"loss": 2.8206,
"step": 21000
},
{
"epoch": 0.32,
"learning_rate": 3.4191176470588236e-05,
"loss": 2.8125,
"step": 21500
},
{
"epoch": 0.32,
"learning_rate": 3.382352941176471e-05,
"loss": 2.8116,
"step": 22000
},
{
"epoch": 0.33,
"learning_rate": 3.345588235294118e-05,
"loss": 2.7593,
"step": 22500
},
{
"epoch": 0.34,
"learning_rate": 3.308823529411765e-05,
"loss": 2.7886,
"step": 23000
},
{
"epoch": 0.35,
"learning_rate": 3.272058823529412e-05,
"loss": 2.7583,
"step": 23500
},
{
"epoch": 0.35,
"learning_rate": 3.235294117647059e-05,
"loss": 2.7966,
"step": 24000
},
{
"epoch": 0.36,
"learning_rate": 3.198529411764706e-05,
"loss": 2.7796,
"step": 24500
},
{
"epoch": 0.37,
"learning_rate": 3.161764705882353e-05,
"loss": 2.7715,
"step": 25000
},
{
"epoch": 0.38,
"learning_rate": 3.125e-05,
"loss": 2.7665,
"step": 25500
},
{
"epoch": 0.38,
"learning_rate": 3.0882352941176475e-05,
"loss": 2.7638,
"step": 26000
},
{
"epoch": 0.39,
"learning_rate": 3.0514705882352945e-05,
"loss": 2.5882,
"step": 26500
},
{
"epoch": 0.4,
"learning_rate": 3.0147058823529413e-05,
"loss": 2.4162,
"step": 27000
},
{
"epoch": 0.4,
"learning_rate": 2.9779411764705883e-05,
"loss": 2.3511,
"step": 27500
},
{
"epoch": 0.41,
"learning_rate": 2.9411764705882354e-05,
"loss": 2.4274,
"step": 28000
},
{
"epoch": 0.42,
"learning_rate": 2.9044117647058828e-05,
"loss": 2.2739,
"step": 28500
},
{
"epoch": 0.43,
"learning_rate": 2.8676470588235295e-05,
"loss": 2.367,
"step": 29000
},
{
"epoch": 0.43,
"learning_rate": 2.8308823529411766e-05,
"loss": 2.343,
"step": 29500
},
{
"epoch": 0.44,
"learning_rate": 2.7941176470588236e-05,
"loss": 2.3481,
"step": 30000
},
{
"epoch": 0.45,
"learning_rate": 2.757352941176471e-05,
"loss": 2.3269,
"step": 30500
},
{
"epoch": 0.46,
"learning_rate": 2.7205882352941174e-05,
"loss": 2.3847,
"step": 31000
},
{
"epoch": 0.46,
"learning_rate": 2.6838235294117648e-05,
"loss": 2.3687,
"step": 31500
},
{
"epoch": 0.47,
"learning_rate": 2.647058823529412e-05,
"loss": 2.3464,
"step": 32000
},
{
"epoch": 0.48,
"learning_rate": 2.6102941176470593e-05,
"loss": 2.3378,
"step": 32500
},
{
"epoch": 0.49,
"learning_rate": 2.5735294117647057e-05,
"loss": 2.3135,
"step": 33000
},
{
"epoch": 0.49,
"learning_rate": 2.536764705882353e-05,
"loss": 2.3938,
"step": 33500
},
{
"epoch": 0.5,
"learning_rate": 2.5e-05,
"loss": 2.34,
"step": 34000
},
{
"epoch": 0.51,
"learning_rate": 2.4632352941176472e-05,
"loss": 2.2955,
"step": 34500
},
{
"epoch": 0.51,
"learning_rate": 2.4264705882352942e-05,
"loss": 2.3362,
"step": 35000
},
{
"epoch": 0.52,
"learning_rate": 2.3897058823529413e-05,
"loss": 2.3094,
"step": 35500
},
{
"epoch": 0.53,
"learning_rate": 2.3529411764705884e-05,
"loss": 2.3167,
"step": 36000
},
{
"epoch": 0.54,
"learning_rate": 2.3161764705882354e-05,
"loss": 2.3323,
"step": 36500
},
{
"epoch": 0.54,
"learning_rate": 2.2794117647058825e-05,
"loss": 2.3169,
"step": 37000
},
{
"epoch": 0.55,
"learning_rate": 2.2426470588235296e-05,
"loss": 2.2706,
"step": 37500
},
{
"epoch": 0.56,
"learning_rate": 2.2058823529411766e-05,
"loss": 2.288,
"step": 38000
},
{
"epoch": 0.57,
"learning_rate": 2.1691176470588237e-05,
"loss": 2.2829,
"step": 38500
},
{
"epoch": 0.57,
"learning_rate": 2.1323529411764707e-05,
"loss": 2.2698,
"step": 39000
},
{
"epoch": 0.58,
"learning_rate": 2.0955882352941178e-05,
"loss": 2.2751,
"step": 39500
},
{
"epoch": 0.59,
"learning_rate": 2.058823529411765e-05,
"loss": 2.282,
"step": 40000
},
{
"epoch": 0.6,
"learning_rate": 2.022058823529412e-05,
"loss": 2.332,
"step": 40500
},
{
"epoch": 0.6,
"learning_rate": 1.9852941176470586e-05,
"loss": 2.2739,
"step": 41000
},
{
"epoch": 0.61,
"learning_rate": 1.948529411764706e-05,
"loss": 2.1936,
"step": 41500
},
{
"epoch": 0.62,
"learning_rate": 1.9117647058823528e-05,
"loss": 2.3159,
"step": 42000
},
{
"epoch": 0.62,
"learning_rate": 1.8750000000000002e-05,
"loss": 2.2772,
"step": 42500
},
{
"epoch": 0.63,
"learning_rate": 1.8382352941176472e-05,
"loss": 2.2725,
"step": 43000
},
{
"epoch": 0.64,
"learning_rate": 1.8014705882352943e-05,
"loss": 2.2081,
"step": 43500
},
{
"epoch": 0.65,
"learning_rate": 1.7647058823529414e-05,
"loss": 2.2688,
"step": 44000
},
{
"epoch": 0.65,
"learning_rate": 1.7279411764705884e-05,
"loss": 2.2481,
"step": 44500
},
{
"epoch": 0.66,
"learning_rate": 1.6911764705882355e-05,
"loss": 2.2502,
"step": 45000
},
{
"epoch": 0.67,
"learning_rate": 1.6544117647058825e-05,
"loss": 2.2903,
"step": 45500
},
{
"epoch": 0.68,
"learning_rate": 1.6176470588235296e-05,
"loss": 2.2717,
"step": 46000
},
{
"epoch": 0.68,
"learning_rate": 1.5808823529411763e-05,
"loss": 2.2524,
"step": 46500
},
{
"epoch": 0.69,
"learning_rate": 1.5441176470588237e-05,
"loss": 2.2076,
"step": 47000
},
{
"epoch": 0.7,
"learning_rate": 1.5073529411764706e-05,
"loss": 2.1851,
"step": 47500
},
{
"epoch": 0.71,
"learning_rate": 1.4705882352941177e-05,
"loss": 2.2289,
"step": 48000
},
{
"epoch": 0.71,
"learning_rate": 1.4338235294117647e-05,
"loss": 2.2356,
"step": 48500
},
{
"epoch": 0.72,
"learning_rate": 1.3970588235294118e-05,
"loss": 2.1841,
"step": 49000
},
{
"epoch": 0.73,
"learning_rate": 1.3602941176470587e-05,
"loss": 2.2053,
"step": 49500
},
{
"epoch": 0.74,
"learning_rate": 1.323529411764706e-05,
"loss": 2.168,
"step": 50000
},
{
"epoch": 0.74,
"learning_rate": 1.2867647058823528e-05,
"loss": 2.2052,
"step": 50500
},
{
"epoch": 0.75,
"learning_rate": 1.25e-05,
"loss": 2.5832,
"step": 51000
},
{
"epoch": 0.76,
"learning_rate": 1.2132352941176471e-05,
"loss": 2.8124,
"step": 51500
},
{
"epoch": 0.76,
"learning_rate": 1.1764705882352942e-05,
"loss": 2.8186,
"step": 52000
},
{
"epoch": 0.77,
"learning_rate": 1.1397058823529412e-05,
"loss": 2.8211,
"step": 52500
},
{
"epoch": 0.78,
"learning_rate": 1.1029411764705883e-05,
"loss": 2.7935,
"step": 53000
},
{
"epoch": 0.79,
"learning_rate": 1.0661764705882354e-05,
"loss": 2.773,
"step": 53500
},
{
"epoch": 0.79,
"learning_rate": 1.0294117647058824e-05,
"loss": 2.7405,
"step": 54000
},
{
"epoch": 0.8,
"learning_rate": 9.926470588235293e-06,
"loss": 2.7454,
"step": 54500
},
{
"epoch": 0.81,
"learning_rate": 9.558823529411764e-06,
"loss": 2.7332,
"step": 55000
},
{
"epoch": 0.82,
"learning_rate": 9.191176470588236e-06,
"loss": 2.7391,
"step": 55500
},
{
"epoch": 0.82,
"learning_rate": 8.823529411764707e-06,
"loss": 2.6863,
"step": 56000
},
{
"epoch": 0.83,
"learning_rate": 8.455882352941177e-06,
"loss": 2.6596,
"step": 56500
},
{
"epoch": 0.84,
"learning_rate": 8.088235294117648e-06,
"loss": 2.6591,
"step": 57000
},
{
"epoch": 0.85,
"learning_rate": 7.720588235294119e-06,
"loss": 2.6312,
"step": 57500
},
{
"epoch": 0.85,
"learning_rate": 7.3529411764705884e-06,
"loss": 2.6305,
"step": 58000
},
{
"epoch": 0.86,
"learning_rate": 6.985294117647059e-06,
"loss": 2.6348,
"step": 58500
},
{
"epoch": 0.87,
"learning_rate": 6.61764705882353e-06,
"loss": 2.643,
"step": 59000
},
{
"epoch": 0.88,
"learning_rate": 6.25e-06,
"loss": 2.6259,
"step": 59500
},
{
"epoch": 0.88,
"learning_rate": 5.882352941176471e-06,
"loss": 2.6413,
"step": 60000
},
{
"epoch": 0.89,
"learning_rate": 5.5147058823529415e-06,
"loss": 2.613,
"step": 60500
},
{
"epoch": 0.9,
"learning_rate": 5.147058823529412e-06,
"loss": 2.6375,
"step": 61000
},
{
"epoch": 0.9,
"learning_rate": 4.779411764705882e-06,
"loss": 2.6253,
"step": 61500
},
{
"epoch": 0.91,
"learning_rate": 4.411764705882353e-06,
"loss": 2.6316,
"step": 62000
},
{
"epoch": 0.92,
"learning_rate": 4.044117647058824e-06,
"loss": 2.6303,
"step": 62500
},
{
"epoch": 0.93,
"learning_rate": 3.6764705882352942e-06,
"loss": 2.5973,
"step": 63000
},
{
"epoch": 0.93,
"learning_rate": 3.308823529411765e-06,
"loss": 2.6142,
"step": 63500
},
{
"epoch": 0.94,
"learning_rate": 2.9411764705882355e-06,
"loss": 2.617,
"step": 64000
},
{
"epoch": 0.95,
"learning_rate": 2.573529411764706e-06,
"loss": 2.6031,
"step": 64500
},
{
"epoch": 0.96,
"learning_rate": 2.2058823529411767e-06,
"loss": 2.6227,
"step": 65000
},
{
"epoch": 0.96,
"learning_rate": 1.8382352941176471e-06,
"loss": 2.5721,
"step": 65500
},
{
"epoch": 0.97,
"learning_rate": 1.4705882352941177e-06,
"loss": 2.6024,
"step": 66000
},
{
"epoch": 0.98,
"learning_rate": 1.1029411764705884e-06,
"loss": 2.6035,
"step": 66500
},
{
"epoch": 0.99,
"learning_rate": 7.352941176470589e-07,
"loss": 2.8269,
"step": 67000
},
{
"epoch": 0.99,
"learning_rate": 3.6764705882352943e-07,
"loss": 2.7742,
"step": 67500
},
{
"epoch": 1.0,
"learning_rate": 0.0,
"loss": 2.7087,
"step": 68000
}
],
"logging_steps": 500,
"max_steps": 68000,
"num_input_tokens_seen": 0,
"num_train_epochs": 9223372036854775807,
"save_steps": 2000,
"total_flos": 1.137142923264e+18,
"train_batch_size": 32,
"trial_name": null,
"trial_params": null
}