|
{ |
|
"best_metric": 0.9904761904761905, |
|
"best_model_checkpoint": "distilhubert-finetuned-cry-detector/checkpoint-341", |
|
"epoch": 3.994143484626647, |
|
"eval_steps": 500, |
|
"global_step": 341, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.9956076134699854, |
|
"eval_accuracy": 0.9787545787545787, |
|
"eval_f1": 0.9788275754377449, |
|
"eval_loss": 0.07363971322774887, |
|
"eval_precision": 0.9790028856592324, |
|
"eval_recall": 0.9787545787545787, |
|
"eval_runtime": 4.1234, |
|
"eval_samples_per_second": 331.041, |
|
"eval_steps_per_second": 41.471, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 1.9912152269399708, |
|
"eval_accuracy": 0.9758241758241758, |
|
"eval_f1": 0.9760458977669253, |
|
"eval_loss": 0.0680176243185997, |
|
"eval_precision": 0.9770364666252971, |
|
"eval_recall": 0.9758241758241758, |
|
"eval_runtime": 3.9805, |
|
"eval_samples_per_second": 342.924, |
|
"eval_steps_per_second": 42.96, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 2.998535871156662, |
|
"eval_accuracy": 0.9875457875457876, |
|
"eval_f1": 0.987550637300988, |
|
"eval_loss": 0.044689420610666275, |
|
"eval_precision": 0.9875567820840008, |
|
"eval_recall": 0.9875457875457876, |
|
"eval_runtime": 4.1011, |
|
"eval_samples_per_second": 332.839, |
|
"eval_steps_per_second": 41.696, |
|
"step": 256 |
|
}, |
|
{ |
|
"epoch": 3.994143484626647, |
|
"eval_accuracy": 0.9904761904761905, |
|
"eval_f1": 0.9904945509377323, |
|
"eval_loss": 0.045228052884340286, |
|
"eval_precision": 0.9905452314533213, |
|
"eval_recall": 0.9904761904761905, |
|
"eval_runtime": 3.9393, |
|
"eval_samples_per_second": 346.509, |
|
"eval_steps_per_second": 43.409, |
|
"step": 341 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 595, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 7, |
|
"save_steps": 500, |
|
"stateful_callbacks": { |
|
"EarlyStoppingCallback": { |
|
"args": { |
|
"early_stopping_patience": 3, |
|
"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": false |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 4.966803039744e+16, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|