|
{ |
|
"best_metric": 0.5653966665267944, |
|
"best_model_checkpoint": "/mnt/beegfs/farid/mlora/outputs/xnli/aya-101/sw/rank4_lr5e-5/checkpoint-5500", |
|
"epoch": 0.24445893089960888, |
|
"eval_steps": 500, |
|
"global_step": 6000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.020371577574967405, |
|
"grad_norm": 2.0201494693756104, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 1.1486, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.020371577574967405, |
|
"eval_accuracy": 0.37309236947791163, |
|
"eval_f1": 0.3322283925001309, |
|
"eval_loss": 1.0934925079345703, |
|
"eval_runtime": 413.7386, |
|
"eval_samples_per_second": 6.018, |
|
"eval_steps_per_second": 0.377, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.04074315514993481, |
|
"grad_norm": 5.576170444488525, |
|
"learning_rate": 4.62962962962963e-05, |
|
"loss": 1.0336, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.04074315514993481, |
|
"eval_accuracy": 0.6514056224899598, |
|
"eval_f1": 0.6496455979119086, |
|
"eval_loss": 0.8499080538749695, |
|
"eval_runtime": 411.5812, |
|
"eval_samples_per_second": 6.05, |
|
"eval_steps_per_second": 0.379, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.06111473272490222, |
|
"grad_norm": 7.646302700042725, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 0.8636, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.06111473272490222, |
|
"eval_accuracy": 0.7269076305220884, |
|
"eval_f1": 0.7287323684689043, |
|
"eval_loss": 0.668684184551239, |
|
"eval_runtime": 411.6805, |
|
"eval_samples_per_second": 6.048, |
|
"eval_steps_per_second": 0.379, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.08148631029986962, |
|
"grad_norm": 11.417831420898438, |
|
"learning_rate": 3.7037037037037037e-05, |
|
"loss": 0.7968, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.08148631029986962, |
|
"eval_accuracy": 0.734136546184739, |
|
"eval_f1": 0.7373689502738993, |
|
"eval_loss": 0.6560496091842651, |
|
"eval_runtime": 411.618, |
|
"eval_samples_per_second": 6.049, |
|
"eval_steps_per_second": 0.379, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.10185788787483703, |
|
"grad_norm": 5.362828254699707, |
|
"learning_rate": 3.240740740740741e-05, |
|
"loss": 0.7744, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.10185788787483703, |
|
"eval_accuracy": 0.7546184738955823, |
|
"eval_f1": 0.7561677733171569, |
|
"eval_loss": 0.6078478693962097, |
|
"eval_runtime": 411.4482, |
|
"eval_samples_per_second": 6.052, |
|
"eval_steps_per_second": 0.379, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.12222946544980444, |
|
"grad_norm": 5.366226673126221, |
|
"learning_rate": 2.777777777777778e-05, |
|
"loss": 0.7559, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.12222946544980444, |
|
"eval_accuracy": 0.7570281124497992, |
|
"eval_f1": 0.7592593531306621, |
|
"eval_loss": 0.6063372492790222, |
|
"eval_runtime": 411.1805, |
|
"eval_samples_per_second": 6.056, |
|
"eval_steps_per_second": 0.379, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.14260104302477183, |
|
"grad_norm": 7.356509685516357, |
|
"learning_rate": 2.314814814814815e-05, |
|
"loss": 0.7419, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.14260104302477183, |
|
"eval_accuracy": 0.7734939759036145, |
|
"eval_f1": 0.7749800378134201, |
|
"eval_loss": 0.5820600390434265, |
|
"eval_runtime": 411.2963, |
|
"eval_samples_per_second": 6.054, |
|
"eval_steps_per_second": 0.379, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.16297262059973924, |
|
"grad_norm": 6.393350124359131, |
|
"learning_rate": 1.8518518518518518e-05, |
|
"loss": 0.7382, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.16297262059973924, |
|
"eval_accuracy": 0.7746987951807229, |
|
"eval_f1": 0.7750757863908069, |
|
"eval_loss": 0.5675287842750549, |
|
"eval_runtime": 411.5669, |
|
"eval_samples_per_second": 6.05, |
|
"eval_steps_per_second": 0.379, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.18334419817470665, |
|
"grad_norm": 6.975522994995117, |
|
"learning_rate": 1.388888888888889e-05, |
|
"loss": 0.7207, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.18334419817470665, |
|
"eval_accuracy": 0.7738955823293173, |
|
"eval_f1": 0.7758231525059284, |
|
"eval_loss": 0.5766967535018921, |
|
"eval_runtime": 411.2036, |
|
"eval_samples_per_second": 6.055, |
|
"eval_steps_per_second": 0.379, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.20371577574967406, |
|
"grad_norm": 6.244735240936279, |
|
"learning_rate": 9.259259259259259e-06, |
|
"loss": 0.7221, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.20371577574967406, |
|
"eval_accuracy": 0.7819277108433735, |
|
"eval_f1": 0.7828326523350859, |
|
"eval_loss": 0.5711483359336853, |
|
"eval_runtime": 432.5918, |
|
"eval_samples_per_second": 5.756, |
|
"eval_steps_per_second": 0.361, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.22408735332464147, |
|
"grad_norm": 4.810456275939941, |
|
"learning_rate": 4.6296296296296296e-06, |
|
"loss": 0.7079, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.22408735332464147, |
|
"eval_accuracy": 0.7823293172690763, |
|
"eval_f1": 0.783225399873768, |
|
"eval_loss": 0.5653966665267944, |
|
"eval_runtime": 411.5268, |
|
"eval_samples_per_second": 6.051, |
|
"eval_steps_per_second": 0.379, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.24445893089960888, |
|
"grad_norm": 7.606008052825928, |
|
"learning_rate": 0.0, |
|
"loss": 0.7073, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.24445893089960888, |
|
"eval_accuracy": 0.7835341365461848, |
|
"eval_f1": 0.7843248776628492, |
|
"eval_loss": 0.5673787593841553, |
|
"eval_runtime": 411.0953, |
|
"eval_samples_per_second": 6.057, |
|
"eval_steps_per_second": 0.379, |
|
"step": 6000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 6000, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 500, |
|
"total_flos": 8.03166870528e+17, |
|
"train_batch_size": 16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|