bts-7b-881 / checkpoint-132 /trainer_state.json
jspr's picture
initial commit
4672c95
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.962406015037594,
"eval_steps": 17,
"global_step": 132,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02,
"learning_rate": 2e-05,
"loss": 1.8373,
"step": 1
},
{
"epoch": 0.02,
"eval_loss": 1.8333783149719238,
"eval_runtime": 16.5756,
"eval_samples_per_second": 2.715,
"eval_steps_per_second": 1.388,
"step": 1
},
{
"epoch": 0.03,
"learning_rate": 4e-05,
"loss": 1.8119,
"step": 2
},
{
"epoch": 0.05,
"learning_rate": 6e-05,
"loss": 1.8301,
"step": 3
},
{
"epoch": 0.06,
"learning_rate": 8e-05,
"loss": 1.7976,
"step": 4
},
{
"epoch": 0.08,
"learning_rate": 0.0001,
"loss": 1.853,
"step": 5
},
{
"epoch": 0.09,
"learning_rate": 0.00012,
"loss": 1.7586,
"step": 6
},
{
"epoch": 0.11,
"learning_rate": 0.00014,
"loss": 1.8416,
"step": 7
},
{
"epoch": 0.12,
"learning_rate": 0.00016,
"loss": 1.7755,
"step": 8
},
{
"epoch": 0.14,
"learning_rate": 0.00018,
"loss": 1.804,
"step": 9
},
{
"epoch": 0.15,
"learning_rate": 0.0002,
"loss": 1.7858,
"step": 10
},
{
"epoch": 0.17,
"learning_rate": 0.0001999923511388017,
"loss": 1.7311,
"step": 11
},
{
"epoch": 0.18,
"learning_rate": 0.0001999694057253083,
"loss": 1.7012,
"step": 12
},
{
"epoch": 0.2,
"learning_rate": 0.00019993116726964554,
"loss": 1.742,
"step": 13
},
{
"epoch": 0.21,
"learning_rate": 0.00019987764162142613,
"loss": 1.7209,
"step": 14
},
{
"epoch": 0.23,
"learning_rate": 0.0001998088369688552,
"loss": 1.7578,
"step": 15
},
{
"epoch": 0.24,
"learning_rate": 0.00019972476383747748,
"loss": 1.6906,
"step": 16
},
{
"epoch": 0.26,
"learning_rate": 0.0001996254350885672,
"loss": 1.738,
"step": 17
},
{
"epoch": 0.26,
"eval_loss": 1.754616141319275,
"eval_runtime": 16.6372,
"eval_samples_per_second": 2.705,
"eval_steps_per_second": 1.382,
"step": 17
},
{
"epoch": 0.27,
"learning_rate": 0.0001995108659171607,
"loss": 1.7976,
"step": 18
},
{
"epoch": 0.29,
"learning_rate": 0.00019938107384973166,
"loss": 1.7781,
"step": 19
},
{
"epoch": 0.3,
"learning_rate": 0.00019923607874151032,
"loss": 1.7737,
"step": 20
},
{
"epoch": 0.32,
"learning_rate": 0.00019907590277344582,
"loss": 1.6662,
"step": 21
},
{
"epoch": 0.33,
"learning_rate": 0.00019890057044881306,
"loss": 1.7869,
"step": 22
},
{
"epoch": 0.35,
"learning_rate": 0.0001987101085894644,
"loss": 1.6646,
"step": 23
},
{
"epoch": 0.36,
"learning_rate": 0.00019850454633172631,
"loss": 1.7482,
"step": 24
},
{
"epoch": 0.38,
"learning_rate": 0.0001982839151219424,
"loss": 1.7112,
"step": 25
},
{
"epoch": 0.39,
"learning_rate": 0.00019804824871166255,
"loss": 1.7083,
"step": 26
},
{
"epoch": 0.41,
"learning_rate": 0.00019779758315248004,
"loss": 1.728,
"step": 27
},
{
"epoch": 0.42,
"learning_rate": 0.00019753195679051628,
"loss": 1.7292,
"step": 28
},
{
"epoch": 0.44,
"learning_rate": 0.0001972514102605547,
"loss": 1.6902,
"step": 29
},
{
"epoch": 0.45,
"learning_rate": 0.00019695598647982468,
"loss": 1.7432,
"step": 30
},
{
"epoch": 0.47,
"learning_rate": 0.00019664573064143604,
"loss": 1.7258,
"step": 31
},
{
"epoch": 0.48,
"learning_rate": 0.00019632069020746572,
"loss": 1.7363,
"step": 32
},
{
"epoch": 0.5,
"learning_rate": 0.00019598091490169694,
"loss": 1.7142,
"step": 33
},
{
"epoch": 0.51,
"learning_rate": 0.00019562645670201276,
"loss": 1.704,
"step": 34
},
{
"epoch": 0.51,
"eval_loss": 1.7388739585876465,
"eval_runtime": 16.5955,
"eval_samples_per_second": 2.712,
"eval_steps_per_second": 1.386,
"step": 34
},
{
"epoch": 0.53,
"learning_rate": 0.0001952573698324446,
"loss": 1.7393,
"step": 35
},
{
"epoch": 0.54,
"learning_rate": 0.00019487371075487713,
"loss": 1.7568,
"step": 36
},
{
"epoch": 0.56,
"learning_rate": 0.000194475538160411,
"loss": 1.7445,
"step": 37
},
{
"epoch": 0.57,
"learning_rate": 0.0001940629129603844,
"loss": 1.6595,
"step": 38
},
{
"epoch": 0.59,
"learning_rate": 0.00019363589827705492,
"loss": 1.7288,
"step": 39
},
{
"epoch": 0.6,
"learning_rate": 0.00019319455943394347,
"loss": 1.6342,
"step": 40
},
{
"epoch": 0.62,
"learning_rate": 0.00019273896394584103,
"loss": 1.7066,
"step": 41
},
{
"epoch": 0.63,
"learning_rate": 0.00019226918150848068,
"loss": 1.6558,
"step": 42
},
{
"epoch": 0.65,
"learning_rate": 0.00019178528398787551,
"loss": 1.6756,
"step": 43
},
{
"epoch": 0.66,
"learning_rate": 0.00019128734540932495,
"loss": 1.7146,
"step": 44
},
{
"epoch": 0.68,
"learning_rate": 0.00019077544194609042,
"loss": 1.7043,
"step": 45
},
{
"epoch": 0.69,
"learning_rate": 0.00019024965190774263,
"loss": 1.6396,
"step": 46
},
{
"epoch": 0.71,
"learning_rate": 0.00018971005572818213,
"loss": 1.648,
"step": 47
},
{
"epoch": 0.72,
"learning_rate": 0.00018915673595333444,
"loss": 1.5988,
"step": 48
},
{
"epoch": 0.74,
"learning_rate": 0.00018858977722852275,
"loss": 1.7394,
"step": 49
},
{
"epoch": 0.75,
"learning_rate": 0.00018800926628551886,
"loss": 1.6362,
"step": 50
},
{
"epoch": 0.77,
"learning_rate": 0.00018741529192927526,
"loss": 1.6762,
"step": 51
},
{
"epoch": 0.77,
"eval_loss": 1.7409569025039673,
"eval_runtime": 16.7147,
"eval_samples_per_second": 2.692,
"eval_steps_per_second": 1.376,
"step": 51
},
{
"epoch": 0.78,
"learning_rate": 0.00018680794502434018,
"loss": 1.6534,
"step": 52
},
{
"epoch": 0.8,
"learning_rate": 0.00018618731848095706,
"loss": 1.6551,
"step": 53
},
{
"epoch": 0.81,
"learning_rate": 0.00018555350724085162,
"loss": 1.6297,
"step": 54
},
{
"epoch": 0.83,
"learning_rate": 0.0001849066082627079,
"loss": 1.7152,
"step": 55
},
{
"epoch": 0.84,
"learning_rate": 0.00018424672050733576,
"loss": 1.7062,
"step": 56
},
{
"epoch": 0.86,
"learning_rate": 0.00018357394492253215,
"loss": 1.5742,
"step": 57
},
{
"epoch": 0.87,
"learning_rate": 0.00018288838442763838,
"loss": 1.6424,
"step": 58
},
{
"epoch": 0.89,
"learning_rate": 0.00018219014389779585,
"loss": 1.6544,
"step": 59
},
{
"epoch": 0.9,
"learning_rate": 0.00018147933014790244,
"loss": 1.6179,
"step": 60
},
{
"epoch": 0.92,
"learning_rate": 0.0001807560519162724,
"loss": 1.6823,
"step": 61
},
{
"epoch": 0.93,
"learning_rate": 0.00018002041984800174,
"loss": 1.5845,
"step": 62
},
{
"epoch": 0.95,
"learning_rate": 0.00017927254647804209,
"loss": 1.6177,
"step": 63
},
{
"epoch": 0.96,
"learning_rate": 0.0001785125462139855,
"loss": 1.6196,
"step": 64
},
{
"epoch": 0.98,
"learning_rate": 0.00017774053531856258,
"loss": 1.6526,
"step": 65
},
{
"epoch": 0.99,
"learning_rate": 0.000176956631891857,
"loss": 1.5792,
"step": 66
},
{
"epoch": 1.01,
"learning_rate": 0.00017616095585323878,
"loss": 1.6652,
"step": 67
},
{
"epoch": 1.02,
"learning_rate": 0.00017535362892301954,
"loss": 1.5981,
"step": 68
},
{
"epoch": 1.02,
"eval_loss": 1.7487449645996094,
"eval_runtime": 16.6837,
"eval_samples_per_second": 2.697,
"eval_steps_per_second": 1.379,
"step": 68
},
{
"epoch": 1.02,
"learning_rate": 0.0001745347746038319,
"loss": 1.6053,
"step": 69
},
{
"epoch": 1.03,
"learning_rate": 0.0001737045181617364,
"loss": 1.6171,
"step": 70
},
{
"epoch": 1.05,
"learning_rate": 0.00017286298660705875,
"loss": 1.631,
"step": 71
},
{
"epoch": 1.06,
"learning_rate": 0.00017201030867496005,
"loss": 1.5558,
"step": 72
},
{
"epoch": 1.08,
"learning_rate": 0.00017114661480574332,
"loss": 1.5339,
"step": 73
},
{
"epoch": 1.09,
"learning_rate": 0.000170272037124899,
"loss": 1.6548,
"step": 74
},
{
"epoch": 1.11,
"learning_rate": 0.00016938670942289293,
"loss": 1.5526,
"step": 75
},
{
"epoch": 1.12,
"learning_rate": 0.00016849076713469914,
"loss": 1.5565,
"step": 76
},
{
"epoch": 1.14,
"learning_rate": 0.00016758434731908178,
"loss": 1.5604,
"step": 77
},
{
"epoch": 1.15,
"learning_rate": 0.00016666758863762793,
"loss": 1.5654,
"step": 78
},
{
"epoch": 1.17,
"learning_rate": 0.00016574063133353582,
"loss": 1.5967,
"step": 79
},
{
"epoch": 1.18,
"learning_rate": 0.00016480361721016054,
"loss": 1.5608,
"step": 80
},
{
"epoch": 1.2,
"learning_rate": 0.00016385668960932143,
"loss": 1.5801,
"step": 81
},
{
"epoch": 1.21,
"learning_rate": 0.00016289999338937427,
"loss": 1.5472,
"step": 82
},
{
"epoch": 1.23,
"learning_rate": 0.00016193367490305088,
"loss": 1.5126,
"step": 83
},
{
"epoch": 1.24,
"learning_rate": 0.00016095788197507081,
"loss": 1.5896,
"step": 84
},
{
"epoch": 1.26,
"learning_rate": 0.00015997276387952732,
"loss": 1.5593,
"step": 85
},
{
"epoch": 1.26,
"eval_loss": 1.79562509059906,
"eval_runtime": 16.6149,
"eval_samples_per_second": 2.708,
"eval_steps_per_second": 1.384,
"step": 85
},
{
"epoch": 1.27,
"learning_rate": 0.00015897847131705195,
"loss": 1.5616,
"step": 86
},
{
"epoch": 1.29,
"learning_rate": 0.00015797515639176074,
"loss": 1.596,
"step": 87
},
{
"epoch": 1.3,
"learning_rate": 0.0001569629725879857,
"loss": 1.595,
"step": 88
},
{
"epoch": 1.32,
"learning_rate": 0.00015594207474679532,
"loss": 1.5442,
"step": 89
},
{
"epoch": 1.33,
"learning_rate": 0.00015491261904230727,
"loss": 1.5812,
"step": 90
},
{
"epoch": 1.35,
"learning_rate": 0.00015387476295779736,
"loss": 1.5214,
"step": 91
},
{
"epoch": 1.36,
"learning_rate": 0.00015282866526160837,
"loss": 1.5403,
"step": 92
},
{
"epoch": 1.38,
"learning_rate": 0.0001517744859828618,
"loss": 1.4751,
"step": 93
},
{
"epoch": 1.39,
"learning_rate": 0.00015071238638697732,
"loss": 1.5861,
"step": 94
},
{
"epoch": 1.41,
"learning_rate": 0.00014964252895100264,
"loss": 1.5334,
"step": 95
},
{
"epoch": 1.42,
"learning_rate": 0.00014856507733875836,
"loss": 1.4884,
"step": 96
},
{
"epoch": 1.44,
"learning_rate": 0.00014748019637580114,
"loss": 1.5297,
"step": 97
},
{
"epoch": 1.45,
"learning_rate": 0.00014638805202420895,
"loss": 1.4725,
"step": 98
},
{
"epoch": 1.47,
"learning_rate": 0.0001452888113571929,
"loss": 1.4849,
"step": 99
},
{
"epoch": 1.48,
"learning_rate": 0.0001441826425335387,
"loss": 1.5654,
"step": 100
},
{
"epoch": 1.5,
"learning_rate": 0.00014306971477188223,
"loss": 1.6349,
"step": 101
},
{
"epoch": 1.51,
"learning_rate": 0.0001419501983248229,
"loss": 1.4415,
"step": 102
},
{
"epoch": 1.51,
"eval_loss": 1.7860450744628906,
"eval_runtime": 16.6303,
"eval_samples_per_second": 2.706,
"eval_steps_per_second": 1.383,
"step": 102
},
{
"epoch": 1.53,
"learning_rate": 0.00014082426445287903,
"loss": 1.5735,
"step": 103
},
{
"epoch": 1.54,
"learning_rate": 0.00013969208539828872,
"loss": 1.5328,
"step": 104
},
{
"epoch": 1.56,
"learning_rate": 0.00013855383435866077,
"loss": 1.5466,
"step": 105
},
{
"epoch": 1.57,
"learning_rate": 0.00013740968546047935,
"loss": 1.5861,
"step": 106
},
{
"epoch": 1.59,
"learning_rate": 0.0001362598137324667,
"loss": 1.5449,
"step": 107
},
{
"epoch": 1.6,
"learning_rate": 0.00013510439507880776,
"loss": 1.6084,
"step": 108
},
{
"epoch": 1.62,
"learning_rate": 0.00013394360625224068,
"loss": 1.5249,
"step": 109
},
{
"epoch": 1.63,
"learning_rate": 0.00013277762482701767,
"loss": 1.5594,
"step": 110
},
{
"epoch": 1.65,
"learning_rate": 0.00013160662917174044,
"loss": 1.5877,
"step": 111
},
{
"epoch": 1.66,
"learning_rate": 0.0001304307984220736,
"loss": 1.4964,
"step": 112
},
{
"epoch": 1.68,
"learning_rate": 0.0001292503124533411,
"loss": 1.4538,
"step": 113
},
{
"epoch": 1.69,
"learning_rate": 0.0001280653518530093,
"loss": 1.4454,
"step": 114
},
{
"epoch": 1.71,
"learning_rate": 0.00012687609789306144,
"loss": 1.4512,
"step": 115
},
{
"epoch": 1.72,
"learning_rate": 0.0001256827325022668,
"loss": 1.38,
"step": 116
},
{
"epoch": 1.74,
"learning_rate": 0.00012448543823835015,
"loss": 1.5207,
"step": 117
},
{
"epoch": 1.75,
"learning_rate": 0.00012328439826006415,
"loss": 1.4787,
"step": 118
},
{
"epoch": 1.77,
"learning_rate": 0.0001220797962991706,
"loss": 1.6098,
"step": 119
},
{
"epoch": 1.77,
"eval_loss": 1.8019659519195557,
"eval_runtime": 16.646,
"eval_samples_per_second": 2.703,
"eval_steps_per_second": 1.382,
"step": 119
},
{
"epoch": 1.78,
"learning_rate": 0.00012087181663233354,
"loss": 1.4798,
"step": 120
},
{
"epoch": 1.8,
"learning_rate": 0.00011966064405292887,
"loss": 1.5248,
"step": 121
},
{
"epoch": 1.81,
"learning_rate": 0.0001184464638427756,
"loss": 1.4309,
"step": 122
},
{
"epoch": 1.83,
"learning_rate": 0.00011722946174379168,
"loss": 1.4477,
"step": 123
},
{
"epoch": 1.84,
"learning_rate": 0.00011600982392957978,
"loss": 1.4843,
"step": 124
},
{
"epoch": 1.86,
"learning_rate": 0.00011478773697694691,
"loss": 1.4298,
"step": 125
},
{
"epoch": 1.87,
"learning_rate": 0.00011356338783736255,
"loss": 1.4775,
"step": 126
},
{
"epoch": 1.89,
"learning_rate": 0.0001123369638083593,
"loss": 1.5159,
"step": 127
},
{
"epoch": 1.9,
"learning_rate": 0.00011110865250488047,
"loss": 1.3696,
"step": 128
},
{
"epoch": 1.92,
"learning_rate": 0.00010987864183057943,
"loss": 1.4259,
"step": 129
},
{
"epoch": 1.93,
"learning_rate": 0.00010864711994907458,
"loss": 1.5685,
"step": 130
},
{
"epoch": 1.95,
"learning_rate": 0.00010741427525516463,
"loss": 1.4997,
"step": 131
},
{
"epoch": 1.96,
"learning_rate": 0.00010618029634600843,
"loss": 1.4718,
"step": 132
}
],
"logging_steps": 1,
"max_steps": 264,
"num_input_tokens_seen": 0,
"num_train_epochs": 4,
"save_steps": 66,
"total_flos": 1.735504576024412e+17,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}