emailgen-pythia-410m-deduped / trainer_state.json
pszemraj's picture
End of training
5bc5d25
raw
history blame
10.6 kB
{
"best_metric": 0.6156604453386092,
"best_model_checkpoint": "./runtime-ar/checkpoint-152",
"epoch": 3.9950900163666123,
"global_step": 152,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05,
"learning_rate": 5e-05,
"loss": 3.3108,
"step": 2
},
{
"epoch": 0.1,
"learning_rate": 0.0001,
"loss": 4.8935,
"step": 4
},
{
"epoch": 0.16,
"learning_rate": 0.00015000000000000001,
"loss": 3.6355,
"step": 6
},
{
"epoch": 0.21,
"learning_rate": 0.0002,
"loss": 3.4513,
"step": 8
},
{
"epoch": 0.26,
"learning_rate": 0.0001999048221581858,
"loss": 3.2178,
"step": 10
},
{
"epoch": 0.31,
"learning_rate": 0.00019961946980917456,
"loss": 3.0584,
"step": 12
},
{
"epoch": 0.37,
"learning_rate": 0.00019914448613738106,
"loss": 2.908,
"step": 14
},
{
"epoch": 0.42,
"learning_rate": 0.00019848077530122083,
"loss": 2.7828,
"step": 16
},
{
"epoch": 0.47,
"learning_rate": 0.00019762960071199333,
"loss": 2.6914,
"step": 18
},
{
"epoch": 0.52,
"learning_rate": 0.00019659258262890683,
"loss": 2.6232,
"step": 20
},
{
"epoch": 0.58,
"learning_rate": 0.0001953716950748227,
"loss": 2.5839,
"step": 22
},
{
"epoch": 0.63,
"learning_rate": 0.00019396926207859084,
"loss": 2.5405,
"step": 24
},
{
"epoch": 0.68,
"learning_rate": 0.0001923879532511287,
"loss": 2.4946,
"step": 26
},
{
"epoch": 0.73,
"learning_rate": 0.000190630778703665,
"loss": 2.4245,
"step": 28
},
{
"epoch": 0.79,
"learning_rate": 0.00018870108331782217,
"loss": 2.4099,
"step": 30
},
{
"epoch": 0.84,
"learning_rate": 0.00018660254037844388,
"loss": 2.3972,
"step": 32
},
{
"epoch": 0.89,
"learning_rate": 0.0001843391445812886,
"loss": 2.3371,
"step": 34
},
{
"epoch": 0.94,
"learning_rate": 0.0001819152044288992,
"loss": 2.276,
"step": 36
},
{
"epoch": 1.0,
"learning_rate": 0.00017933533402912354,
"loss": 2.3067,
"step": 38
},
{
"epoch": 1.0,
"eval_accuracy": 0.5594373263429553,
"eval_loss": 2.3559441566467285,
"eval_runtime": 87.3704,
"eval_samples_per_second": 6.192,
"eval_steps_per_second": 3.102,
"step": 38
},
{
"epoch": 1.05,
"learning_rate": 0.0001766044443118978,
"loss": 2.3629,
"step": 40
},
{
"epoch": 1.1,
"learning_rate": 0.0001737277336810124,
"loss": 2.0942,
"step": 42
},
{
"epoch": 1.16,
"learning_rate": 0.00017071067811865476,
"loss": 2.0975,
"step": 44
},
{
"epoch": 1.21,
"learning_rate": 0.00016755902076156604,
"loss": 2.0554,
"step": 46
},
{
"epoch": 1.26,
"learning_rate": 0.00016427876096865394,
"loss": 2.0336,
"step": 48
},
{
"epoch": 1.31,
"learning_rate": 0.00016087614290087208,
"loss": 2.0584,
"step": 50
},
{
"epoch": 1.37,
"learning_rate": 0.0001573576436351046,
"loss": 1.9959,
"step": 52
},
{
"epoch": 1.42,
"learning_rate": 0.0001537299608346824,
"loss": 1.9855,
"step": 54
},
{
"epoch": 1.47,
"learning_rate": 0.00015000000000000001,
"loss": 1.9941,
"step": 56
},
{
"epoch": 1.52,
"learning_rate": 0.00014617486132350343,
"loss": 1.9589,
"step": 58
},
{
"epoch": 1.58,
"learning_rate": 0.00014226182617406996,
"loss": 1.9918,
"step": 60
},
{
"epoch": 1.63,
"learning_rate": 0.000138268343236509,
"loss": 1.9673,
"step": 62
},
{
"epoch": 1.68,
"learning_rate": 0.00013420201433256689,
"loss": 1.9626,
"step": 64
},
{
"epoch": 1.73,
"learning_rate": 0.00013007057995042732,
"loss": 1.9327,
"step": 66
},
{
"epoch": 1.79,
"learning_rate": 0.00012588190451025207,
"loss": 1.9334,
"step": 68
},
{
"epoch": 1.84,
"learning_rate": 0.00012164396139381029,
"loss": 1.9405,
"step": 70
},
{
"epoch": 1.89,
"learning_rate": 0.00011736481776669306,
"loss": 1.8394,
"step": 72
},
{
"epoch": 1.94,
"learning_rate": 0.00011305261922200519,
"loss": 1.9284,
"step": 74
},
{
"epoch": 2.0,
"learning_rate": 0.00010871557427476583,
"loss": 1.9243,
"step": 76
},
{
"epoch": 2.0,
"eval_accuracy": 0.5974633090939628,
"eval_loss": 2.1282806396484375,
"eval_runtime": 88.3059,
"eval_samples_per_second": 6.126,
"eval_steps_per_second": 3.069,
"step": 76
},
{
"epoch": 2.05,
"learning_rate": 0.00010436193873653361,
"loss": 1.8745,
"step": 78
},
{
"epoch": 2.1,
"learning_rate": 0.0001,
"loss": 1.6456,
"step": 80
},
{
"epoch": 2.16,
"learning_rate": 9.563806126346642e-05,
"loss": 1.6496,
"step": 82
},
{
"epoch": 2.21,
"learning_rate": 9.128442572523417e-05,
"loss": 1.6571,
"step": 84
},
{
"epoch": 2.26,
"learning_rate": 8.694738077799488e-05,
"loss": 1.6366,
"step": 86
},
{
"epoch": 2.31,
"learning_rate": 8.263518223330697e-05,
"loss": 1.6033,
"step": 88
},
{
"epoch": 2.37,
"learning_rate": 7.835603860618972e-05,
"loss": 1.6042,
"step": 90
},
{
"epoch": 2.42,
"learning_rate": 7.411809548974792e-05,
"loss": 1.6111,
"step": 92
},
{
"epoch": 2.47,
"learning_rate": 6.992942004957271e-05,
"loss": 1.6058,
"step": 94
},
{
"epoch": 2.52,
"learning_rate": 6.579798566743314e-05,
"loss": 1.6079,
"step": 96
},
{
"epoch": 2.58,
"learning_rate": 6.173165676349103e-05,
"loss": 1.5789,
"step": 98
},
{
"epoch": 2.63,
"learning_rate": 5.773817382593008e-05,
"loss": 1.6267,
"step": 100
},
{
"epoch": 2.68,
"learning_rate": 5.382513867649663e-05,
"loss": 1.5811,
"step": 102
},
{
"epoch": 2.73,
"learning_rate": 5.000000000000002e-05,
"loss": 1.5605,
"step": 104
},
{
"epoch": 2.79,
"learning_rate": 4.6270039165317605e-05,
"loss": 1.5828,
"step": 106
},
{
"epoch": 2.84,
"learning_rate": 4.264235636489542e-05,
"loss": 1.5579,
"step": 108
},
{
"epoch": 2.89,
"learning_rate": 3.9123857099127936e-05,
"loss": 1.5991,
"step": 110
},
{
"epoch": 2.94,
"learning_rate": 3.5721239031346066e-05,
"loss": 1.5641,
"step": 112
},
{
"epoch": 3.0,
"learning_rate": 3.244097923843398e-05,
"loss": 1.5971,
"step": 114
},
{
"epoch": 3.0,
"eval_accuracy": 0.6139908093264839,
"eval_loss": 2.075925827026367,
"eval_runtime": 88.4185,
"eval_samples_per_second": 6.119,
"eval_steps_per_second": 3.065,
"step": 114
},
{
"epoch": 3.05,
"learning_rate": 2.9289321881345254e-05,
"loss": 1.5486,
"step": 116
},
{
"epoch": 3.1,
"learning_rate": 2.6272266318987603e-05,
"loss": 1.4341,
"step": 118
},
{
"epoch": 3.16,
"learning_rate": 2.339555568810221e-05,
"loss": 1.4085,
"step": 120
},
{
"epoch": 3.21,
"learning_rate": 2.0664665970876496e-05,
"loss": 1.3819,
"step": 122
},
{
"epoch": 3.26,
"learning_rate": 1.808479557110081e-05,
"loss": 1.3935,
"step": 124
},
{
"epoch": 3.31,
"learning_rate": 1.566085541871145e-05,
"loss": 1.4073,
"step": 126
},
{
"epoch": 3.37,
"learning_rate": 1.339745962155613e-05,
"loss": 1.419,
"step": 128
},
{
"epoch": 3.42,
"learning_rate": 1.129891668217783e-05,
"loss": 1.3807,
"step": 130
},
{
"epoch": 3.47,
"learning_rate": 9.369221296335006e-06,
"loss": 1.3881,
"step": 132
},
{
"epoch": 3.52,
"learning_rate": 7.612046748871327e-06,
"loss": 1.3674,
"step": 134
},
{
"epoch": 3.58,
"learning_rate": 6.030737921409169e-06,
"loss": 1.3847,
"step": 136
},
{
"epoch": 3.63,
"learning_rate": 4.628304925177318e-06,
"loss": 1.3734,
"step": 138
},
{
"epoch": 3.68,
"learning_rate": 3.40741737109318e-06,
"loss": 1.3969,
"step": 140
},
{
"epoch": 3.73,
"learning_rate": 2.3703992880066638e-06,
"loss": 1.3544,
"step": 142
},
{
"epoch": 3.79,
"learning_rate": 1.5192246987791981e-06,
"loss": 1.4021,
"step": 144
},
{
"epoch": 3.84,
"learning_rate": 8.555138626189618e-07,
"loss": 1.4061,
"step": 146
},
{
"epoch": 3.89,
"learning_rate": 3.805301908254455e-07,
"loss": 1.3621,
"step": 148
},
{
"epoch": 3.94,
"learning_rate": 9.517784181422019e-08,
"loss": 1.3759,
"step": 150
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"loss": 1.4139,
"step": 152
},
{
"epoch": 4.0,
"eval_accuracy": 0.6156604453386092,
"eval_loss": 2.101821184158325,
"eval_runtime": 88.2947,
"eval_samples_per_second": 6.127,
"eval_steps_per_second": 3.069,
"step": 152
},
{
"epoch": 4.0,
"step": 152,
"total_flos": 8.481201929060352e+16,
"train_loss": 1.970301656346572,
"train_runtime": 9467.2073,
"train_samples_per_second": 2.062,
"train_steps_per_second": 0.016
}
],
"max_steps": 152,
"num_train_epochs": 4,
"total_flos": 8.481201929060352e+16,
"trial_name": null,
"trial_params": null
}