sa_sub2 / trainer_state.json
researchaccount's picture
Added files
ed51040
{
"best_metric": 0.8211337852223485,
"best_model_checkpoint": "./train/run-25/checkpoint-1375",
"epoch": 1.0,
"global_step": 1375,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.36,
"learning_rate": 2.1231225195028302e-05,
"loss": 0.4399,
"step": 500
},
{
"epoch": 0.73,
"learning_rate": 5.177088990820726e-06,
"loss": 0.3702,
"step": 1000
},
{
"epoch": 1.0,
"eval_accuracy": 0.8641818181818182,
"eval_average_recall": 0.8178683917059567,
"eval_loss": 0.33707520365715027,
"eval_macro_f1": 0.8211337852223485,
"eval_macro_f1_neg": 0.7398609501738123,
"eval_macro_f1_pos": 0.8204693242861183,
"eval_macro_f1_pos_neg": 0.7801651372299653,
"eval_rec_neg": 0.7239229024943311,
"eval_rec_neu": 0.9071199143468951,
"eval_rec_pos": 0.822562358276644,
"eval_runtime": 24.4173,
"eval_samples_per_second": 450.499,
"step": 1375
}
],
"max_steps": 1375,
"num_train_epochs": 1,
"total_flos": 0,
"trial_name": null,
"trial_params": {
"learning_rate": 3.0000000000000004e-05,
"num_train_epochs": 1,
"seed": 12345
}
}