| { |
| "best_global_step": 465, |
| "best_metric": 0.7806512878301866, |
| "best_model_checkpoint": "/kaggle/working/transformer_output/checkpoint-465", |
| "epoch": 3.0, |
| "eval_steps": 500, |
| "global_step": 465, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.3225806451612903, |
| "grad_norm": 3.0797810554504395, |
| "learning_rate": 1.2727272727272728e-05, |
| "loss": 1.1631031799316407, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.6451612903225806, |
| "grad_norm": 3.368655204772949, |
| "learning_rate": 1.9951016547387286e-05, |
| "loss": 0.6313877105712891, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.967741935483871, |
| "grad_norm": 5.035543918609619, |
| "learning_rate": 1.947950097086825e-05, |
| "loss": 0.5252903747558594, |
| "step": 150 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.8930589184826473, |
| "eval_f1": 0.900084594871237, |
| "eval_hate_f1": 0.5027322404371585, |
| "eval_hate_prec": 0.4125560538116592, |
| "eval_hate_rec": 0.6433566433566433, |
| "eval_loss": 0.2638593912124634, |
| "eval_macro_f1": 0.7711109967956045, |
| "eval_precision": 0.9107537030140248, |
| "eval_recall": 0.8930589184826473, |
| "eval_runtime": 4.8064, |
| "eval_samples_per_second": 515.563, |
| "eval_steps_per_second": 8.114, |
| "step": 155 |
| }, |
| { |
| "epoch": 1.2903225806451613, |
| "grad_norm": 2.4748122692108154, |
| "learning_rate": 1.8529927230890757e-05, |
| "loss": 0.501370506286621, |
| "step": 200 |
| }, |
| { |
| "epoch": 1.6129032258064515, |
| "grad_norm": 2.85589599609375, |
| "learning_rate": 1.715018302529852e-05, |
| "loss": 0.4371611404418945, |
| "step": 250 |
| }, |
| { |
| "epoch": 1.935483870967742, |
| "grad_norm": 2.5343403816223145, |
| "learning_rate": 1.5409849862969994e-05, |
| "loss": 0.44288436889648436, |
| "step": 300 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_accuracy": 0.8930589184826473, |
| "eval_f1": 0.9015939212097996, |
| "eval_hate_f1": 0.49604221635883905, |
| "eval_hate_prec": 0.3983050847457627, |
| "eval_hate_rec": 0.6573426573426573, |
| "eval_loss": 0.24266110360622406, |
| "eval_macro_f1": 0.7724936604193823, |
| "eval_precision": 0.9157473422011047, |
| "eval_recall": 0.8930589184826473, |
| "eval_runtime": 4.9035, |
| "eval_samples_per_second": 505.358, |
| "eval_steps_per_second": 7.954, |
| "step": 310 |
| }, |
| { |
| "epoch": 2.258064516129032, |
| "grad_norm": 6.783071994781494, |
| "learning_rate": 1.3396694017461708e-05, |
| "loss": 0.4083076477050781, |
| "step": 350 |
| }, |
| { |
| "epoch": 2.5806451612903225, |
| "grad_norm": 4.064752578735352, |
| "learning_rate": 1.1212240409675825e-05, |
| "loss": 0.36163642883300784, |
| "step": 400 |
| }, |
| { |
| "epoch": 2.903225806451613, |
| "grad_norm": 3.120911121368408, |
| "learning_rate": 8.966652631884506e-06, |
| "loss": 0.3937630081176758, |
| "step": 450 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_accuracy": 0.8970944309927361, |
| "eval_f1": 0.9050220867632944, |
| "eval_hate_f1": 0.5053191489361702, |
| "eval_hate_prec": 0.40772532188841204, |
| "eval_hate_rec": 0.6643356643356644, |
| "eval_loss": 0.23981328308582306, |
| "eval_macro_f1": 0.7806512878301866, |
| "eval_precision": 0.9177721181109427, |
| "eval_recall": 0.8970944309927361, |
| "eval_runtime": 4.8801, |
| "eval_samples_per_second": 507.772, |
| "eval_steps_per_second": 7.992, |
| "step": 465 |
| } |
| ], |
| "logging_steps": 50, |
| "max_steps": 775, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 5, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "EarlyStoppingCallback": { |
| "args": { |
| "early_stopping_patience": 2, |
| "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": 1467136861839936.0, |
| "train_batch_size": 64, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
