| { | |
| "best_metric": 1.614416241645813, | |
| "best_model_checkpoint": "distilbert_weight_decay_final_metrics/checkpoint-6872", | |
| "epoch": 8.0, | |
| "global_step": 13744, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 1.970896391152503e-05, | |
| "loss": 0.0266, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 1.941792782305006e-05, | |
| "loss": 0.0377, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 1.9126891734575087e-05, | |
| "loss": 0.0319, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.7770232031692134, | |
| "eval_f1": 0.781143699897714, | |
| "eval_loss": 1.8788982629776, | |
| "eval_runtime": 8.0848, | |
| "eval_samples_per_second": 437.119, | |
| "eval_steps_per_second": 27.335, | |
| "step": 1718 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "learning_rate": 1.8835855646100117e-05, | |
| "loss": 0.0387, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.46, | |
| "learning_rate": 1.8544819557625148e-05, | |
| "loss": 0.0449, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.75, | |
| "learning_rate": 1.825378346915018e-05, | |
| "loss": 0.0541, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.7744765138653085, | |
| "eval_f1": 0.778056670029498, | |
| "eval_loss": 1.7166885137557983, | |
| "eval_runtime": 8.1001, | |
| "eval_samples_per_second": 436.288, | |
| "eval_steps_per_second": 27.283, | |
| "step": 3436 | |
| }, | |
| { | |
| "epoch": 2.04, | |
| "learning_rate": 1.7962747380675206e-05, | |
| "loss": 0.0534, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "learning_rate": 1.7671711292200233e-05, | |
| "loss": 0.0394, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 2.62, | |
| "learning_rate": 1.7380675203725264e-05, | |
| "loss": 0.0573, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 2.91, | |
| "learning_rate": 1.708963911525029e-05, | |
| "loss": 0.0438, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.7756083757781551, | |
| "eval_f1": 0.7790011542334092, | |
| "eval_loss": 1.7141435146331787, | |
| "eval_runtime": 8.0923, | |
| "eval_samples_per_second": 436.712, | |
| "eval_steps_per_second": 27.31, | |
| "step": 5154 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "learning_rate": 1.679860302677532e-05, | |
| "loss": 0.0331, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 3.49, | |
| "learning_rate": 1.650756693830035e-05, | |
| "loss": 0.0402, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 3.78, | |
| "learning_rate": 1.621653084982538e-05, | |
| "loss": 0.0388, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.7761743067345784, | |
| "eval_f1": 0.7801920701860957, | |
| "eval_loss": 1.614416241645813, | |
| "eval_runtime": 8.1981, | |
| "eval_samples_per_second": 431.077, | |
| "eval_steps_per_second": 26.958, | |
| "step": 6872 | |
| }, | |
| { | |
| "epoch": 4.07, | |
| "learning_rate": 1.592549476135041e-05, | |
| "loss": 0.0458, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 4.37, | |
| "learning_rate": 1.5634458672875437e-05, | |
| "loss": 0.0282, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 4.66, | |
| "learning_rate": 1.5343422584400468e-05, | |
| "loss": 0.0348, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 4.95, | |
| "learning_rate": 1.5052386495925495e-05, | |
| "loss": 0.0319, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.7849462365591398, | |
| "eval_f1": 0.7885570926799784, | |
| "eval_loss": 1.7275830507278442, | |
| "eval_runtime": 8.0966, | |
| "eval_samples_per_second": 436.48, | |
| "eval_steps_per_second": 27.295, | |
| "step": 8590 | |
| }, | |
| { | |
| "epoch": 5.24, | |
| "learning_rate": 1.4761350407450526e-05, | |
| "loss": 0.0338, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 5.53, | |
| "learning_rate": 1.4470314318975553e-05, | |
| "loss": 0.0272, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 5.82, | |
| "learning_rate": 1.4179278230500584e-05, | |
| "loss": 0.0269, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.7795698924731183, | |
| "eval_f1": 0.7833736396261405, | |
| "eval_loss": 1.8119807243347168, | |
| "eval_runtime": 8.1074, | |
| "eval_samples_per_second": 435.896, | |
| "eval_steps_per_second": 27.259, | |
| "step": 10308 | |
| }, | |
| { | |
| "epoch": 6.11, | |
| "learning_rate": 1.3888242142025613e-05, | |
| "loss": 0.0267, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 6.4, | |
| "learning_rate": 1.3597206053550642e-05, | |
| "loss": 0.0191, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 6.69, | |
| "learning_rate": 1.330616996507567e-05, | |
| "loss": 0.023, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 6.98, | |
| "learning_rate": 1.3015133876600698e-05, | |
| "loss": 0.0251, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.7826825127334465, | |
| "eval_f1": 0.7861199960648936, | |
| "eval_loss": 1.7902283668518066, | |
| "eval_runtime": 8.1049, | |
| "eval_samples_per_second": 436.031, | |
| "eval_steps_per_second": 27.267, | |
| "step": 12026 | |
| }, | |
| { | |
| "epoch": 7.28, | |
| "learning_rate": 1.2724097788125728e-05, | |
| "loss": 0.0239, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 7.57, | |
| "learning_rate": 1.2433061699650757e-05, | |
| "loss": 0.0261, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 7.86, | |
| "learning_rate": 1.2142025611175788e-05, | |
| "loss": 0.0194, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.7869269949066214, | |
| "eval_f1": 0.7901736955407076, | |
| "eval_loss": 1.923709750175476, | |
| "eval_runtime": 9.3451, | |
| "eval_samples_per_second": 378.167, | |
| "eval_steps_per_second": 23.649, | |
| "step": 13744 | |
| } | |
| ], | |
| "max_steps": 34360, | |
| "num_train_epochs": 20, | |
| "total_flos": 6256940082766560.0, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |