parrot-llama-2-13B-epoch3 / trainer_state.json
douy's picture
Upload folder using huggingface_hub
5dffd25
raw
history blame
10.9 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 10,
"global_step": 81,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"learning_rate": 0.0,
"loss": 1.0172,
"step": 1
},
{
"epoch": 0.07,
"learning_rate": 3.1546487678572864e-05,
"loss": 2.3198,
"step": 2
},
{
"epoch": 0.11,
"learning_rate": 5e-05,
"loss": 2.4795,
"step": 3
},
{
"epoch": 0.15,
"learning_rate": 5e-05,
"loss": 1.551,
"step": 4
},
{
"epoch": 0.19,
"learning_rate": 5e-05,
"loss": 1.2034,
"step": 5
},
{
"epoch": 0.22,
"learning_rate": 5e-05,
"loss": 1.0527,
"step": 6
},
{
"epoch": 0.26,
"learning_rate": 5e-05,
"loss": 0.9743,
"step": 7
},
{
"epoch": 0.3,
"learning_rate": 5e-05,
"loss": 0.9738,
"step": 8
},
{
"epoch": 0.33,
"learning_rate": 5e-05,
"loss": 0.939,
"step": 9
},
{
"epoch": 0.37,
"learning_rate": 5e-05,
"loss": 0.955,
"step": 10
},
{
"epoch": 0.37,
"eval_accuracy": 0.2181755564165483,
"eval_loss": 0.9061336517333984,
"eval_runtime": 44.217,
"eval_samples_per_second": 0.814,
"eval_steps_per_second": 0.045,
"step": 10
},
{
"epoch": 0.41,
"learning_rate": 5e-05,
"loss": 0.9513,
"step": 11
},
{
"epoch": 0.44,
"learning_rate": 5e-05,
"loss": 0.94,
"step": 12
},
{
"epoch": 0.48,
"learning_rate": 5e-05,
"loss": 0.9211,
"step": 13
},
{
"epoch": 0.52,
"learning_rate": 5e-05,
"loss": 0.8762,
"step": 14
},
{
"epoch": 0.56,
"learning_rate": 5e-05,
"loss": 0.9013,
"step": 15
},
{
"epoch": 0.59,
"learning_rate": 5e-05,
"loss": 0.8758,
"step": 16
},
{
"epoch": 0.63,
"learning_rate": 5e-05,
"loss": 0.8919,
"step": 17
},
{
"epoch": 0.67,
"learning_rate": 5e-05,
"loss": 0.8589,
"step": 18
},
{
"epoch": 0.7,
"learning_rate": 5e-05,
"loss": 0.8775,
"step": 19
},
{
"epoch": 0.74,
"learning_rate": 5e-05,
"loss": 0.8899,
"step": 20
},
{
"epoch": 0.74,
"eval_accuracy": 0.21962848163933016,
"eval_loss": 0.8734426498413086,
"eval_runtime": 44.8031,
"eval_samples_per_second": 0.804,
"eval_steps_per_second": 0.045,
"step": 20
},
{
"epoch": 0.78,
"learning_rate": 5e-05,
"loss": 0.9011,
"step": 21
},
{
"epoch": 0.81,
"learning_rate": 5e-05,
"loss": 0.9195,
"step": 22
},
{
"epoch": 0.85,
"learning_rate": 5e-05,
"loss": 0.815,
"step": 23
},
{
"epoch": 0.89,
"learning_rate": 5e-05,
"loss": 0.8216,
"step": 24
},
{
"epoch": 0.93,
"learning_rate": 5e-05,
"loss": 0.8559,
"step": 25
},
{
"epoch": 0.96,
"learning_rate": 5e-05,
"loss": 0.9071,
"step": 26
},
{
"epoch": 1.0,
"learning_rate": 5e-05,
"loss": 0.6228,
"step": 27
},
{
"epoch": 1.04,
"learning_rate": 5e-05,
"loss": 0.3934,
"step": 28
},
{
"epoch": 1.07,
"learning_rate": 5e-05,
"loss": 0.8846,
"step": 29
},
{
"epoch": 1.11,
"learning_rate": 5e-05,
"loss": 0.879,
"step": 30
},
{
"epoch": 1.11,
"eval_accuracy": 0.2173899005553403,
"eval_loss": 0.9090863466262817,
"eval_runtime": 45.3228,
"eval_samples_per_second": 0.794,
"eval_steps_per_second": 0.044,
"step": 30
},
{
"epoch": 1.15,
"learning_rate": 5e-05,
"loss": 0.8277,
"step": 31
},
{
"epoch": 1.19,
"learning_rate": 5e-05,
"loss": 0.6593,
"step": 32
},
{
"epoch": 1.22,
"learning_rate": 5e-05,
"loss": 0.5255,
"step": 33
},
{
"epoch": 1.26,
"learning_rate": 5e-05,
"loss": 0.4102,
"step": 34
},
{
"epoch": 1.3,
"learning_rate": 5e-05,
"loss": 0.5839,
"step": 35
},
{
"epoch": 1.33,
"learning_rate": 5e-05,
"loss": 0.6197,
"step": 36
},
{
"epoch": 1.37,
"learning_rate": 5e-05,
"loss": 0.4026,
"step": 37
},
{
"epoch": 1.41,
"learning_rate": 5e-05,
"loss": 0.3835,
"step": 38
},
{
"epoch": 1.44,
"learning_rate": 5e-05,
"loss": 0.3433,
"step": 39
},
{
"epoch": 1.48,
"learning_rate": 5e-05,
"loss": 0.3295,
"step": 40
},
{
"epoch": 1.48,
"eval_accuracy": 0.21730380128287916,
"eval_loss": 0.9803322553634644,
"eval_runtime": 44.8946,
"eval_samples_per_second": 0.802,
"eval_steps_per_second": 0.045,
"step": 40
},
{
"epoch": 1.52,
"learning_rate": 5e-05,
"loss": 0.3101,
"step": 41
},
{
"epoch": 1.56,
"learning_rate": 5e-05,
"loss": 0.3492,
"step": 42
},
{
"epoch": 1.59,
"learning_rate": 5e-05,
"loss": 0.2991,
"step": 43
},
{
"epoch": 1.63,
"learning_rate": 5e-05,
"loss": 0.3412,
"step": 44
},
{
"epoch": 1.67,
"learning_rate": 5e-05,
"loss": 0.3824,
"step": 45
},
{
"epoch": 1.7,
"learning_rate": 5e-05,
"loss": 0.3694,
"step": 46
},
{
"epoch": 1.74,
"learning_rate": 5e-05,
"loss": 0.3926,
"step": 47
},
{
"epoch": 1.78,
"learning_rate": 5e-05,
"loss": 0.3993,
"step": 48
},
{
"epoch": 1.81,
"learning_rate": 5e-05,
"loss": 0.3678,
"step": 49
},
{
"epoch": 1.85,
"learning_rate": 5e-05,
"loss": 0.3711,
"step": 50
},
{
"epoch": 1.85,
"eval_accuracy": 0.2173576133281674,
"eval_loss": 0.9820255637168884,
"eval_runtime": 44.7843,
"eval_samples_per_second": 0.804,
"eval_steps_per_second": 0.045,
"step": 50
},
{
"epoch": 1.89,
"learning_rate": 5e-05,
"loss": 0.3825,
"step": 51
},
{
"epoch": 1.93,
"learning_rate": 5e-05,
"loss": 0.3914,
"step": 52
},
{
"epoch": 1.96,
"learning_rate": 5e-05,
"loss": 0.3962,
"step": 53
},
{
"epoch": 2.0,
"learning_rate": 5e-05,
"loss": 0.2342,
"step": 54
},
{
"epoch": 2.04,
"learning_rate": 5e-05,
"loss": 0.1572,
"step": 55
},
{
"epoch": 2.07,
"learning_rate": 5e-05,
"loss": 0.4409,
"step": 56
},
{
"epoch": 2.11,
"learning_rate": 5e-05,
"loss": 0.4072,
"step": 57
},
{
"epoch": 2.15,
"learning_rate": 5e-05,
"loss": 0.4306,
"step": 58
},
{
"epoch": 2.19,
"learning_rate": 5e-05,
"loss": 0.338,
"step": 59
},
{
"epoch": 2.22,
"learning_rate": 5e-05,
"loss": 0.2927,
"step": 60
},
{
"epoch": 2.22,
"eval_accuracy": 0.2152804683800422,
"eval_loss": 1.0269818305969238,
"eval_runtime": 44.7279,
"eval_samples_per_second": 0.805,
"eval_steps_per_second": 0.045,
"step": 60
},
{
"epoch": 2.26,
"learning_rate": 5e-05,
"loss": 0.3043,
"step": 61
},
{
"epoch": 2.3,
"learning_rate": 5e-05,
"loss": 0.3134,
"step": 62
},
{
"epoch": 2.33,
"learning_rate": 5e-05,
"loss": 0.2565,
"step": 63
},
{
"epoch": 2.37,
"learning_rate": 5e-05,
"loss": 0.2333,
"step": 64
},
{
"epoch": 2.41,
"learning_rate": 5e-05,
"loss": 0.2291,
"step": 65
},
{
"epoch": 2.44,
"learning_rate": 5e-05,
"loss": 0.1783,
"step": 66
},
{
"epoch": 2.48,
"learning_rate": 5e-05,
"loss": 0.2058,
"step": 67
},
{
"epoch": 2.52,
"learning_rate": 5e-05,
"loss": 0.208,
"step": 68
},
{
"epoch": 2.56,
"learning_rate": 5e-05,
"loss": 0.1745,
"step": 69
},
{
"epoch": 2.59,
"learning_rate": 5e-05,
"loss": 0.1703,
"step": 70
},
{
"epoch": 2.59,
"eval_accuracy": 0.2131279865685135,
"eval_loss": 1.0966472625732422,
"eval_runtime": 44.8433,
"eval_samples_per_second": 0.803,
"eval_steps_per_second": 0.045,
"step": 70
},
{
"epoch": 2.63,
"learning_rate": 5e-05,
"loss": 0.2235,
"step": 71
},
{
"epoch": 2.67,
"learning_rate": 5e-05,
"loss": 0.2062,
"step": 72
},
{
"epoch": 2.7,
"learning_rate": 5e-05,
"loss": 0.2094,
"step": 73
},
{
"epoch": 2.74,
"learning_rate": 5e-05,
"loss": 0.2123,
"step": 74
},
{
"epoch": 2.78,
"learning_rate": 5e-05,
"loss": 0.2144,
"step": 75
},
{
"epoch": 2.81,
"learning_rate": 5e-05,
"loss": 0.2091,
"step": 76
},
{
"epoch": 2.85,
"learning_rate": 5e-05,
"loss": 0.2116,
"step": 77
},
{
"epoch": 2.89,
"learning_rate": 5e-05,
"loss": 0.2123,
"step": 78
},
{
"epoch": 2.93,
"learning_rate": 5e-05,
"loss": 0.1891,
"step": 79
},
{
"epoch": 2.96,
"learning_rate": 5e-05,
"loss": 0.2011,
"step": 80
},
{
"epoch": 2.96,
"eval_accuracy": 0.2144840501097766,
"eval_loss": 1.1488189697265625,
"eval_runtime": 44.8519,
"eval_samples_per_second": 0.803,
"eval_steps_per_second": 0.045,
"step": 80
},
{
"epoch": 3.0,
"learning_rate": 5e-05,
"loss": 0.1493,
"step": 81
},
{
"epoch": 3.0,
"step": 81,
"total_flos": 10840779079680.0,
"train_loss": 0.5814771251178082,
"train_runtime": 6697.7067,
"train_samples_per_second": 0.377,
"train_steps_per_second": 0.012
}
],
"logging_steps": 1.0,
"max_steps": 81,
"num_train_epochs": 3,
"save_steps": 500,
"total_flos": 10840779079680.0,
"trial_name": null,
"trial_params": null
}