uk-mt5-large-gec / trainer_state.json
SashkoMolodec
initial commit
fc922cf
{
"best_metric": 0.23396648466587067,
"best_model_checkpoint": "uk-mt5-small-gec/checkpoint-400",
"epoch": 1.5209125475285172,
"global_step": 400,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.76,
"learning_rate": 0.0008479087452471484,
"loss": 0.8294,
"step": 200
},
{
"epoch": 0.76,
"eval_google_bleu": 0.4205046652337586,
"eval_loss": 0.24924862384796143,
"eval_runtime": 330.8079,
"eval_samples_per_second": 5.653,
"eval_steps_per_second": 0.354,
"step": 200
},
{
"epoch": 1.52,
"learning_rate": 0.0006958174904942966,
"loss": 0.2123,
"step": 400
},
{
"epoch": 1.52,
"eval_google_bleu": 0.42486486486486486,
"eval_loss": 0.23396648466587067,
"eval_runtime": 331.3322,
"eval_samples_per_second": 5.644,
"eval_steps_per_second": 0.353,
"step": 400
}
],
"max_steps": 1315,
"num_train_epochs": 5,
"total_flos": 8901903215468544.0,
"trial_name": null,
"trial_params": null
}