|
{ |
|
"best_metric": 0.9875457875457876, |
|
"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.9663003663003663, |
|
"eval_f1": 0.9661425019393345, |
|
"eval_loss": 0.10625720024108887, |
|
"eval_precision": 0.966167198143384, |
|
"eval_recall": 0.9663003663003663, |
|
"eval_runtime": 6.2148, |
|
"eval_samples_per_second": 219.638, |
|
"eval_steps_per_second": 27.515, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 1.9912152269399708, |
|
"eval_accuracy": 0.9824175824175824, |
|
"eval_f1": 0.9824569651195038, |
|
"eval_loss": 0.06455090641975403, |
|
"eval_precision": 0.9825418778671476, |
|
"eval_recall": 0.9824175824175824, |
|
"eval_runtime": 5.7635, |
|
"eval_samples_per_second": 236.835, |
|
"eval_steps_per_second": 29.669, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 2.998535871156662, |
|
"eval_accuracy": 0.9868131868131869, |
|
"eval_f1": 0.9868427238396278, |
|
"eval_loss": 0.04568345472216606, |
|
"eval_precision": 0.98691808355097, |
|
"eval_recall": 0.9868131868131869, |
|
"eval_runtime": 6.1234, |
|
"eval_samples_per_second": 222.916, |
|
"eval_steps_per_second": 27.926, |
|
"step": 256 |
|
}, |
|
{ |
|
"epoch": 3.994143484626647, |
|
"eval_accuracy": 0.9875457875457876, |
|
"eval_f1": 0.9875962963533335, |
|
"eval_loss": 0.049539219588041306, |
|
"eval_precision": 0.9877998892958556, |
|
"eval_recall": 0.9875457875457876, |
|
"eval_runtime": 6.1353, |
|
"eval_samples_per_second": 222.481, |
|
"eval_steps_per_second": 27.871, |
|
"step": 341 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 850, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 10, |
|
"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 |
|
} |
|
|