PetBERT_mortality / trainer_state.json
seanfarrell's picture
Upload 11 files
d3fa8cd verified
raw
history blame
1.61 kB
{
"best_metric": 0.9956709956709957,
"best_model_checkpoint": "dead_alive_classifier/models/checkpoint-96",
"epoch": 3.0,
"eval_steps": 500,
"global_step": 96,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_accuracy": 0.7878787878787878,
"eval_f1": 0.7878787878787878,
"eval_loss": 0.5285314321517944,
"eval_precision": 0.7878787878787878,
"eval_recall": 0.7878787878787878,
"eval_runtime": 1.1267,
"eval_samples_per_second": 205.029,
"eval_steps_per_second": 13.314,
"step": 32
},
{
"epoch": 2.0,
"eval_accuracy": 0.974025974025974,
"eval_f1": 0.974025974025974,
"eval_loss": 0.2521773874759674,
"eval_precision": 0.974025974025974,
"eval_recall": 0.974025974025974,
"eval_runtime": 1.1532,
"eval_samples_per_second": 200.307,
"eval_steps_per_second": 13.007,
"step": 64
},
{
"epoch": 3.0,
"eval_accuracy": 0.9956709956709957,
"eval_f1": 0.9956709956709957,
"eval_loss": 0.037678323686122894,
"eval_precision": 0.9956709956709957,
"eval_recall": 0.9956709956709957,
"eval_runtime": 1.1364,
"eval_samples_per_second": 203.277,
"eval_steps_per_second": 13.2,
"step": 96
}
],
"logging_steps": 500,
"max_steps": 480,
"num_input_tokens_seen": 0,
"num_train_epochs": 15,
"save_steps": 500,
"total_flos": 395069471843520.0,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}