| { |
| "best_global_step": 60, |
| "best_metric": 0.5067150592803955, |
| "best_model_checkpoint": "tinybert_base_train_kd_mrpc/checkpoint-60", |
| "epoch": 9.0, |
| "eval_steps": 500, |
| "global_step": 135, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 1.0, |
| "grad_norm": 1.0588346719741821, |
| "learning_rate": 4.906666666666667e-05, |
| "loss": 0.6588, |
| "step": 15 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.6666666666666666, |
| "eval_combined_score": 0.7160919540229884, |
| "eval_f1": 0.7655172413793103, |
| "eval_loss": 0.6283461451530457, |
| "eval_runtime": 0.2187, |
| "eval_samples_per_second": 1865.558, |
| "eval_steps_per_second": 9.145, |
| "step": 15 |
| }, |
| { |
| "epoch": 2.0, |
| "grad_norm": 1.6302603483200073, |
| "learning_rate": 4.806666666666667e-05, |
| "loss": 0.5998, |
| "step": 30 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_accuracy": 0.696078431372549, |
| "eval_combined_score": 0.7508605323006945, |
| "eval_f1": 0.8056426332288401, |
| "eval_loss": 0.5731440782546997, |
| "eval_runtime": 0.2338, |
| "eval_samples_per_second": 1744.854, |
| "eval_steps_per_second": 8.553, |
| "step": 30 |
| }, |
| { |
| "epoch": 3.0, |
| "grad_norm": 5.0050811767578125, |
| "learning_rate": 4.706666666666667e-05, |
| "loss": 0.5433, |
| "step": 45 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_accuracy": 0.7401960784313726, |
| "eval_combined_score": 0.786764705882353, |
| "eval_f1": 0.8333333333333334, |
| "eval_loss": 0.5613774657249451, |
| "eval_runtime": 0.2224, |
| "eval_samples_per_second": 1834.58, |
| "eval_steps_per_second": 8.993, |
| "step": 45 |
| }, |
| { |
| "epoch": 4.0, |
| "grad_norm": 10.076091766357422, |
| "learning_rate": 4.606666666666667e-05, |
| "loss": 0.4574, |
| "step": 60 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_accuracy": 0.7916666666666666, |
| "eval_combined_score": 0.8239213197969544, |
| "eval_f1": 0.856175972927242, |
| "eval_loss": 0.5067150592803955, |
| "eval_runtime": 0.2267, |
| "eval_samples_per_second": 1799.599, |
| "eval_steps_per_second": 8.822, |
| "step": 60 |
| }, |
| { |
| "epoch": 5.0, |
| "grad_norm": 12.540026664733887, |
| "learning_rate": 4.5066666666666667e-05, |
| "loss": 0.3525, |
| "step": 75 |
| }, |
| { |
| "epoch": 5.0, |
| "eval_accuracy": 0.7573529411764706, |
| "eval_combined_score": 0.7888398099711753, |
| "eval_f1": 0.8203266787658802, |
| "eval_loss": 0.5203803777694702, |
| "eval_runtime": 0.2188, |
| "eval_samples_per_second": 1864.761, |
| "eval_steps_per_second": 9.141, |
| "step": 75 |
| }, |
| { |
| "epoch": 6.0, |
| "grad_norm": 10.685439109802246, |
| "learning_rate": 4.406666666666667e-05, |
| "loss": 0.3015, |
| "step": 90 |
| }, |
| { |
| "epoch": 6.0, |
| "eval_accuracy": 0.7426470588235294, |
| "eval_combined_score": 0.7904298468231669, |
| "eval_f1": 0.8382126348228043, |
| "eval_loss": 0.829555332660675, |
| "eval_runtime": 0.2251, |
| "eval_samples_per_second": 1812.326, |
| "eval_steps_per_second": 8.884, |
| "step": 90 |
| }, |
| { |
| "epoch": 7.0, |
| "grad_norm": 5.721017360687256, |
| "learning_rate": 4.3066666666666665e-05, |
| "loss": 0.2978, |
| "step": 105 |
| }, |
| { |
| "epoch": 7.0, |
| "eval_accuracy": 0.75, |
| "eval_combined_score": 0.7950626959247649, |
| "eval_f1": 0.8401253918495298, |
| "eval_loss": 0.7863379120826721, |
| "eval_runtime": 0.2298, |
| "eval_samples_per_second": 1775.245, |
| "eval_steps_per_second": 8.702, |
| "step": 105 |
| }, |
| { |
| "epoch": 8.0, |
| "grad_norm": 11.30908203125, |
| "learning_rate": 4.206666666666667e-05, |
| "loss": 0.2163, |
| "step": 120 |
| }, |
| { |
| "epoch": 8.0, |
| "eval_accuracy": 0.7573529411764706, |
| "eval_combined_score": 0.7940610859728507, |
| "eval_f1": 0.8307692307692308, |
| "eval_loss": 0.7096235752105713, |
| "eval_runtime": 0.2225, |
| "eval_samples_per_second": 1833.884, |
| "eval_steps_per_second": 8.99, |
| "step": 120 |
| }, |
| { |
| "epoch": 9.0, |
| "grad_norm": 4.002461910247803, |
| "learning_rate": 4.106666666666667e-05, |
| "loss": 0.1329, |
| "step": 135 |
| }, |
| { |
| "epoch": 9.0, |
| "eval_accuracy": 0.7549019607843137, |
| "eval_combined_score": 0.7881652661064426, |
| "eval_f1": 0.8214285714285714, |
| "eval_loss": 0.7977427840232849, |
| "eval_runtime": 0.2192, |
| "eval_samples_per_second": 1861.282, |
| "eval_steps_per_second": 9.124, |
| "step": 135 |
| }, |
| { |
| "epoch": 9.0, |
| "step": 135, |
| "total_flos": 2186506882215936.0, |
| "train_loss": 0.3955998191127071, |
| "train_runtime": 48.5344, |
| "train_samples_per_second": 3778.761, |
| "train_steps_per_second": 15.453 |
| } |
| ], |
| "logging_steps": 1, |
| "max_steps": 750, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 50, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "EarlyStoppingCallback": { |
| "args": { |
| "early_stopping_patience": 5, |
| "early_stopping_threshold": 0.0 |
| }, |
| "attributes": { |
| "early_stopping_patience_counter": 5 |
| } |
| }, |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": true |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 2186506882215936.0, |
| "train_batch_size": 256, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
