315 lines
6.8 KiB
JSON
315 lines
6.8 KiB
JSON
|
{
|
||
|
|
"best_global_step": null,
|
||
|
|
"best_metric": null,
|
||
|
|
"best_model_checkpoint": null,
|
||
|
|
"epoch": 0.13333333333333333,
|
||
|
|
"eval_steps": 20,
|
||
|
|
"global_step": 40,
|
||
|
|
"is_hyper_param_search": false,
|
||
|
|
"is_local_process_zero": true,
|
||
|
|
"is_world_process_zero": true,
|
||
|
|
"log_history": [
|
||
|
|
{
|
||
|
|
"epoch": 0.0033333333333333335,
|
||
|
|
"grad_norm": 8.0,
|
||
|
|
"learning_rate": 0.0,
|
||
|
|
"loss": 3.704,
|
||
|
|
"step": 1
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.006666666666666667,
|
||
|
|
"grad_norm": 7.5625,
|
||
|
|
"learning_rate": 2.0000000000000003e-06,
|
||
|
|
"loss": 3.753,
|
||
|
|
"step": 2
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.01,
|
||
|
|
"grad_norm": 7.96875,
|
||
|
|
"learning_rate": 4.000000000000001e-06,
|
||
|
|
"loss": 3.7435,
|
||
|
|
"step": 3
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.013333333333333334,
|
||
|
|
"grad_norm": 7.875,
|
||
|
|
"learning_rate": 6e-06,
|
||
|
|
"loss": 3.7596,
|
||
|
|
"step": 4
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.016666666666666666,
|
||
|
|
"grad_norm": 8.25,
|
||
|
|
"learning_rate": 8.000000000000001e-06,
|
||
|
|
"loss": 3.6881,
|
||
|
|
"step": 5
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.02,
|
||
|
|
"grad_norm": 7.96875,
|
||
|
|
"learning_rate": 1e-05,
|
||
|
|
"loss": 3.7124,
|
||
|
|
"step": 6
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.023333333333333334,
|
||
|
|
"grad_norm": 8.1875,
|
||
|
|
"learning_rate": 9.983193277310925e-06,
|
||
|
|
"loss": 3.6898,
|
||
|
|
"step": 7
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.02666666666666667,
|
||
|
|
"grad_norm": 7.3125,
|
||
|
|
"learning_rate": 9.96638655462185e-06,
|
||
|
|
"loss": 3.6003,
|
||
|
|
"step": 8
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.03,
|
||
|
|
"grad_norm": 6.59375,
|
||
|
|
"learning_rate": 9.949579831932774e-06,
|
||
|
|
"loss": 3.3132,
|
||
|
|
"step": 9
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.03333333333333333,
|
||
|
|
"grad_norm": 6.875,
|
||
|
|
"learning_rate": 9.932773109243698e-06,
|
||
|
|
"loss": 3.384,
|
||
|
|
"step": 10
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.03666666666666667,
|
||
|
|
"grad_norm": 7.0625,
|
||
|
|
"learning_rate": 9.915966386554622e-06,
|
||
|
|
"loss": 3.2793,
|
||
|
|
"step": 11
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.04,
|
||
|
|
"grad_norm": 6.34375,
|
||
|
|
"learning_rate": 9.899159663865548e-06,
|
||
|
|
"loss": 3.3562,
|
||
|
|
"step": 12
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.043333333333333335,
|
||
|
|
"grad_norm": 5.9375,
|
||
|
|
"learning_rate": 9.882352941176472e-06,
|
||
|
|
"loss": 3.2332,
|
||
|
|
"step": 13
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.04666666666666667,
|
||
|
|
"grad_norm": 5.9375,
|
||
|
|
"learning_rate": 9.865546218487397e-06,
|
||
|
|
"loss": 3.1515,
|
||
|
|
"step": 14
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.05,
|
||
|
|
"grad_norm": 5.46875,
|
||
|
|
"learning_rate": 9.848739495798321e-06,
|
||
|
|
"loss": 3.2211,
|
||
|
|
"step": 15
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.05333333333333334,
|
||
|
|
"grad_norm": 5.75,
|
||
|
|
"learning_rate": 9.831932773109244e-06,
|
||
|
|
"loss": 3.0058,
|
||
|
|
"step": 16
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.056666666666666664,
|
||
|
|
"grad_norm": 4.96875,
|
||
|
|
"learning_rate": 9.815126050420168e-06,
|
||
|
|
"loss": 2.8119,
|
||
|
|
"step": 17
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06,
|
||
|
|
"grad_norm": 5.0,
|
||
|
|
"learning_rate": 9.798319327731092e-06,
|
||
|
|
"loss": 2.9361,
|
||
|
|
"step": 18
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06333333333333334,
|
||
|
|
"grad_norm": 4.5625,
|
||
|
|
"learning_rate": 9.781512605042018e-06,
|
||
|
|
"loss": 2.9393,
|
||
|
|
"step": 19
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06666666666666667,
|
||
|
|
"grad_norm": 4.78125,
|
||
|
|
"learning_rate": 9.764705882352942e-06,
|
||
|
|
"loss": 2.6777,
|
||
|
|
"step": 20
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.07,
|
||
|
|
"grad_norm": 4.4375,
|
||
|
|
"learning_rate": 9.747899159663867e-06,
|
||
|
|
"loss": 2.7683,
|
||
|
|
"step": 21
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.07333333333333333,
|
||
|
|
"grad_norm": 4.28125,
|
||
|
|
"learning_rate": 9.731092436974791e-06,
|
||
|
|
"loss": 2.6016,
|
||
|
|
"step": 22
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.07666666666666666,
|
||
|
|
"grad_norm": 4.5625,
|
||
|
|
"learning_rate": 9.714285714285715e-06,
|
||
|
|
"loss": 2.8662,
|
||
|
|
"step": 23
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08,
|
||
|
|
"grad_norm": 4.0625,
|
||
|
|
"learning_rate": 9.69747899159664e-06,
|
||
|
|
"loss": 2.5337,
|
||
|
|
"step": 24
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08333333333333333,
|
||
|
|
"grad_norm": 3.765625,
|
||
|
|
"learning_rate": 9.680672268907564e-06,
|
||
|
|
"loss": 2.6618,
|
||
|
|
"step": 25
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08666666666666667,
|
||
|
|
"grad_norm": 3.71875,
|
||
|
|
"learning_rate": 9.663865546218488e-06,
|
||
|
|
"loss": 2.5384,
|
||
|
|
"step": 26
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09,
|
||
|
|
"grad_norm": 3.625,
|
||
|
|
"learning_rate": 9.647058823529412e-06,
|
||
|
|
"loss": 2.3644,
|
||
|
|
"step": 27
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09333333333333334,
|
||
|
|
"grad_norm": 3.375,
|
||
|
|
"learning_rate": 9.630252100840337e-06,
|
||
|
|
"loss": 2.5523,
|
||
|
|
"step": 28
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09666666666666666,
|
||
|
|
"grad_norm": 3.5625,
|
||
|
|
"learning_rate": 9.61344537815126e-06,
|
||
|
|
"loss": 2.4416,
|
||
|
|
"step": 29
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.1,
|
||
|
|
"grad_norm": 3.5625,
|
||
|
|
"learning_rate": 9.596638655462187e-06,
|
||
|
|
"loss": 2.3784,
|
||
|
|
"step": 30
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.10333333333333333,
|
||
|
|
"grad_norm": 3.34375,
|
||
|
|
"learning_rate": 9.579831932773111e-06,
|
||
|
|
"loss": 2.2584,
|
||
|
|
"step": 31
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.10666666666666667,
|
||
|
|
"grad_norm": 3.296875,
|
||
|
|
"learning_rate": 9.563025210084035e-06,
|
||
|
|
"loss": 2.3913,
|
||
|
|
"step": 32
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11,
|
||
|
|
"grad_norm": 3.109375,
|
||
|
|
"learning_rate": 9.54621848739496e-06,
|
||
|
|
"loss": 2.2638,
|
||
|
|
"step": 33
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11333333333333333,
|
||
|
|
"grad_norm": 3.03125,
|
||
|
|
"learning_rate": 9.529411764705882e-06,
|
||
|
|
"loss": 2.4333,
|
||
|
|
"step": 34
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11666666666666667,
|
||
|
|
"grad_norm": 2.921875,
|
||
|
|
"learning_rate": 9.512605042016806e-06,
|
||
|
|
"loss": 2.2413,
|
||
|
|
"step": 35
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12,
|
||
|
|
"grad_norm": 2.71875,
|
||
|
|
"learning_rate": 9.49579831932773e-06,
|
||
|
|
"loss": 2.1486,
|
||
|
|
"step": 36
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12333333333333334,
|
||
|
|
"grad_norm": 2.8125,
|
||
|
|
"learning_rate": 9.478991596638657e-06,
|
||
|
|
"loss": 2.2499,
|
||
|
|
"step": 37
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12666666666666668,
|
||
|
|
"grad_norm": 2.625,
|
||
|
|
"learning_rate": 9.462184873949581e-06,
|
||
|
|
"loss": 2.1166,
|
||
|
|
"step": 38
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.13,
|
||
|
|
"grad_norm": 2.5,
|
||
|
|
"learning_rate": 9.445378151260505e-06,
|
||
|
|
"loss": 2.1453,
|
||
|
|
"step": 39
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.13333333333333333,
|
||
|
|
"grad_norm": 2.625,
|
||
|
|
"learning_rate": 9.42857142857143e-06,
|
||
|
|
"loss": 2.2253,
|
||
|
|
"step": 40
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"logging_steps": 1,
|
||
|
|
"max_steps": 600,
|
||
|
|
"num_input_tokens_seen": 0,
|
||
|
|
"num_train_epochs": 2,
|
||
|
|
"save_steps": 20,
|
||
|
|
"stateful_callbacks": {
|
||
|
|
"TrainerControl": {
|
||
|
|
"args": {
|
||
|
|
"should_epoch_stop": false,
|
||
|
|
"should_evaluate": false,
|
||
|
|
"should_log": false,
|
||
|
|
"should_save": true,
|
||
|
|
"should_training_stop": false
|
||
|
|
},
|
||
|
|
"attributes": {}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"total_flos": 2739693278939136.0,
|
||
|
|
"train_batch_size": 2,
|
||
|
|
"trial_name": null,
|
||
|
|
"trial_params": null
|
||
|
|
}
|