| { | |
| "best_metric": 0.6966881607229021, | |
| "best_model_checkpoint": "/kaggle/working/sweep_08/checkpoint-636", | |
| "epoch": 4.0, | |
| "eval_steps": 500, | |
| "global_step": 636, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.31446540880503143, | |
| "grad_norm": 10.8976469039917, | |
| "learning_rate": 9.97322623828648e-06, | |
| "loss": 0.3486, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.6289308176100629, | |
| "grad_norm": 2.1524722576141357, | |
| "learning_rate": 9.303882195448461e-06, | |
| "loss": 0.3024, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.9433962264150944, | |
| "grad_norm": 38.142704010009766, | |
| "learning_rate": 8.634538152610442e-06, | |
| "loss": 0.2815, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.14253897550111358, | |
| "eval_f1_macro": 0.12475633528265107, | |
| "eval_loss": 0.6558064818382263, | |
| "eval_precision_macro": 0.07126948775055679, | |
| "eval_recall_macro": 0.5, | |
| "eval_runtime": 2.6403, | |
| "eval_samples_per_second": 170.055, | |
| "eval_steps_per_second": 5.681, | |
| "step": 159 | |
| }, | |
| { | |
| "epoch": 1.2578616352201257, | |
| "grad_norm": 5.237600326538086, | |
| "learning_rate": 7.965194109772424e-06, | |
| "loss": 0.295, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.5723270440251573, | |
| "grad_norm": 13.672293663024902, | |
| "learning_rate": 7.295850066934404e-06, | |
| "loss": 0.2368, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.8867924528301887, | |
| "grad_norm": 13.05793285369873, | |
| "learning_rate": 6.626506024096386e-06, | |
| "loss": 0.2014, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.6124721603563474, | |
| "eval_f1_macro": 0.5600418975537235, | |
| "eval_loss": 0.2551872134208679, | |
| "eval_precision_macro": 0.6213023625173715, | |
| "eval_recall_macro": 0.7479707792207793, | |
| "eval_runtime": 2.6406, | |
| "eval_samples_per_second": 170.038, | |
| "eval_steps_per_second": 5.681, | |
| "step": 318 | |
| }, | |
| { | |
| "epoch": 2.20125786163522, | |
| "grad_norm": 6.5023651123046875, | |
| "learning_rate": 5.957161981258367e-06, | |
| "loss": 0.1878, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.5157232704402515, | |
| "grad_norm": 23.201416015625, | |
| "learning_rate": 5.287817938420349e-06, | |
| "loss": 0.1756, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.830188679245283, | |
| "grad_norm": 12.81585693359375, | |
| "learning_rate": 4.61847389558233e-06, | |
| "loss": 0.1784, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.643652561247216, | |
| "eval_f1_macro": 0.5815275641921804, | |
| "eval_loss": 0.32544130086898804, | |
| "eval_precision_macro": 0.6241342249820874, | |
| "eval_recall_macro": 0.753125, | |
| "eval_runtime": 2.6269, | |
| "eval_samples_per_second": 170.921, | |
| "eval_steps_per_second": 5.71, | |
| "step": 477 | |
| }, | |
| { | |
| "epoch": 3.1446540880503147, | |
| "grad_norm": 152.0744171142578, | |
| "learning_rate": 3.949129852744311e-06, | |
| "loss": 0.1484, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.459119496855346, | |
| "grad_norm": 15.98477840423584, | |
| "learning_rate": 3.2797858099062922e-06, | |
| "loss": 0.1869, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 3.7735849056603774, | |
| "grad_norm": 54.855499267578125, | |
| "learning_rate": 2.6104417670682734e-06, | |
| "loss": 0.1257, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.8285077951002228, | |
| "eval_f1_macro": 0.6966881607229021, | |
| "eval_loss": 0.30613085627555847, | |
| "eval_precision_macro": 0.6773720349563046, | |
| "eval_recall_macro": 0.7306412337662338, | |
| "eval_runtime": 2.6199, | |
| "eval_samples_per_second": 171.383, | |
| "eval_steps_per_second": 5.725, | |
| "step": 636 | |
| } | |
| ], | |
| "logging_steps": 50, | |
| "max_steps": 795, | |
| "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": 1886354996127984.0, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |