res_nw_egy / trainer_state.json
nlparabic's picture
End of training
2c1903c verified
{
"best_metric": 0.758726954460144,
"best_model_checkpoint": "/home/iais_marenpielka/Bouthaina/res_nw_egy/checkpoint-21321",
"epoch": 3.0,
"eval_steps": 500,
"global_step": 21321,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 2.9758260250091553,
"learning_rate": 3.413380721387061e-05,
"loss": 1.148,
"step": 7107
},
{
"epoch": 1.0,
"eval_bleu": 0.18719204852263469,
"eval_loss": 0.8351342082023621,
"eval_rouge1": 0.5142833565402041,
"eval_rouge2": 0.2505733532313653,
"eval_rougeL": 0.5096825616972613,
"eval_runtime": 169.5172,
"eval_samples_per_second": 83.838,
"eval_steps_per_second": 10.483,
"step": 7107
},
{
"epoch": 2.0,
"grad_norm": 3.232348680496216,
"learning_rate": 1.7066903606935305e-05,
"loss": 0.7647,
"step": 14214
},
{
"epoch": 2.0,
"eval_bleu": 0.2145574961692684,
"eval_loss": 0.7705017924308777,
"eval_rouge1": 0.5551766890409657,
"eval_rouge2": 0.29664921512752485,
"eval_rougeL": 0.5514002756721588,
"eval_runtime": 108.2094,
"eval_samples_per_second": 131.338,
"eval_steps_per_second": 16.422,
"step": 14214
},
{
"epoch": 3.0,
"grad_norm": 2.0039141178131104,
"learning_rate": 0.0,
"loss": 0.6812,
"step": 21321
},
{
"epoch": 3.0,
"eval_bleu": 0.22465329922096197,
"eval_loss": 0.758726954460144,
"eval_rouge1": 0.5688224315923487,
"eval_rouge2": 0.3116501353023259,
"eval_rougeL": 0.56490169958113,
"eval_runtime": 292.0707,
"eval_samples_per_second": 48.659,
"eval_steps_per_second": 6.084,
"step": 21321
},
{
"epoch": 3.0,
"step": 21321,
"total_flos": 1.1141034983424e+16,
"train_loss": 0.8646288613632099,
"train_runtime": 2092.8708,
"train_samples_per_second": 81.492,
"train_steps_per_second": 10.187
}
],
"logging_steps": 500,
"max_steps": 21321,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 5,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 0
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 1.1141034983424e+16,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}