{ "best_metric": 0.2903565764427185, "best_model_checkpoint": "designTrends/checkpoint-36", "epoch": 2.0, "eval_steps": 500, "global_step": 36, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.05555555555555555, "grad_norm": 31.24921989440918, "learning_rate": 8.333333333333334e-06, "loss": 1.6824, "step": 1 }, { "epoch": 0.1111111111111111, "grad_norm": 20.90537452697754, "learning_rate": 1.6666666666666667e-05, "loss": 1.1182, "step": 2 }, { "epoch": 0.16666666666666666, "grad_norm": 37.51194763183594, "learning_rate": 2.5e-05, "loss": 2.1944, "step": 3 }, { "epoch": 0.2222222222222222, "grad_norm": 30.805044174194336, "learning_rate": 3.3333333333333335e-05, "loss": 1.8025, "step": 4 }, { "epoch": 0.2777777777777778, "grad_norm": 15.017728805541992, "learning_rate": 4.166666666666667e-05, "loss": 0.4371, "step": 5 }, { "epoch": 0.3333333333333333, "grad_norm": 37.36845016479492, "learning_rate": 5e-05, "loss": 2.3815, "step": 6 }, { "epoch": 0.3888888888888889, "grad_norm": 36.02885818481445, "learning_rate": 4.8958333333333335e-05, "loss": 1.3463, "step": 7 }, { "epoch": 0.4444444444444444, "grad_norm": 34.15700149536133, "learning_rate": 4.791666666666667e-05, "loss": 2.0606, "step": 8 }, { "epoch": 0.5, "grad_norm": 30.92742347717285, "learning_rate": 4.6875e-05, "loss": 1.2667, "step": 9 }, { "epoch": 0.5555555555555556, "grad_norm": 17.669174194335938, "learning_rate": 4.5833333333333334e-05, "loss": 0.7387, "step": 10 }, { "epoch": 0.6111111111111112, "grad_norm": 21.606388092041016, "learning_rate": 4.4791666666666673e-05, "loss": 1.0753, "step": 11 }, { "epoch": 0.6666666666666666, "grad_norm": 11.96320629119873, "learning_rate": 4.375e-05, "loss": 0.4321, "step": 12 }, { "epoch": 0.7222222222222222, "grad_norm": 28.005523681640625, "learning_rate": 4.270833333333333e-05, "loss": 1.7522, "step": 13 }, { "epoch": 0.7777777777777778, "grad_norm": 13.161625862121582, "learning_rate": 4.166666666666667e-05, "loss": 0.4633, "step": 14 }, { "epoch": 0.8333333333333334, "grad_norm": 30.44942283630371, "learning_rate": 4.0625000000000005e-05, "loss": 1.2798, "step": 15 }, { "epoch": 0.8888888888888888, "grad_norm": 18.832170486450195, "learning_rate": 3.958333333333333e-05, "loss": 0.6956, "step": 16 }, { "epoch": 0.9444444444444444, "grad_norm": 8.636096000671387, "learning_rate": 3.854166666666667e-05, "loss": 0.2504, "step": 17 }, { "epoch": 1.0, "grad_norm": 4.477694511413574, "learning_rate": 3.7500000000000003e-05, "loss": 0.0766, "step": 18 }, { "epoch": 1.0, "eval_accuracy": 0.7777777777777778, "eval_f1_macro": 0.6875, "eval_f1_micro": 0.7777777777777778, "eval_f1_weighted": 0.6944444444444444, "eval_loss": 1.071279525756836, "eval_precision_macro": 0.65, "eval_precision_micro": 0.7777777777777778, "eval_precision_weighted": 0.6444444444444444, "eval_recall_macro": 0.75, "eval_recall_micro": 0.7777777777777778, "eval_recall_weighted": 0.7777777777777778, "eval_runtime": 10.4277, "eval_samples_per_second": 0.863, "eval_steps_per_second": 0.288, "step": 18 }, { "epoch": 1.0555555555555556, "grad_norm": 11.063207626342773, "learning_rate": 3.6458333333333336e-05, "loss": 0.2519, "step": 19 }, { "epoch": 1.1111111111111112, "grad_norm": 11.03740406036377, "learning_rate": 3.541666666666667e-05, "loss": 0.2958, "step": 20 }, { "epoch": 1.1666666666666667, "grad_norm": 1.8288331031799316, "learning_rate": 3.4375e-05, "loss": 0.0406, "step": 21 }, { "epoch": 1.2222222222222223, "grad_norm": 31.19927215576172, "learning_rate": 3.3333333333333335e-05, "loss": 1.0578, "step": 22 }, { "epoch": 1.2777777777777777, "grad_norm": 5.91452693939209, "learning_rate": 3.229166666666667e-05, "loss": 0.1396, "step": 23 }, { "epoch": 1.3333333333333333, "grad_norm": 28.180068969726562, "learning_rate": 3.125e-05, "loss": 0.9014, "step": 24 }, { "epoch": 1.3888888888888888, "grad_norm": 6.710000038146973, "learning_rate": 3.0208333333333334e-05, "loss": 0.1434, "step": 25 }, { "epoch": 1.4444444444444444, "grad_norm": 1.4782997369766235, "learning_rate": 2.916666666666667e-05, "loss": 0.033, "step": 26 }, { "epoch": 1.5, "grad_norm": 10.041507720947266, "learning_rate": 2.8125000000000003e-05, "loss": 0.2318, "step": 27 }, { "epoch": 1.5555555555555556, "grad_norm": 15.513708114624023, "learning_rate": 2.7083333333333332e-05, "loss": 0.4245, "step": 28 }, { "epoch": 1.6111111111111112, "grad_norm": 5.358726978302002, "learning_rate": 2.604166666666667e-05, "loss": 0.1601, "step": 29 }, { "epoch": 1.6666666666666665, "grad_norm": 1.7197539806365967, "learning_rate": 2.5e-05, "loss": 0.042, "step": 30 }, { "epoch": 1.7222222222222223, "grad_norm": 0.5361698865890503, "learning_rate": 2.3958333333333334e-05, "loss": 0.0149, "step": 31 }, { "epoch": 1.7777777777777777, "grad_norm": 4.341958522796631, "learning_rate": 2.2916666666666667e-05, "loss": 0.1039, "step": 32 }, { "epoch": 1.8333333333333335, "grad_norm": 20.135475158691406, "learning_rate": 2.1875e-05, "loss": 0.5532, "step": 33 }, { "epoch": 1.8888888888888888, "grad_norm": 0.5918666124343872, "learning_rate": 2.0833333333333336e-05, "loss": 0.0155, "step": 34 }, { "epoch": 1.9444444444444444, "grad_norm": 2.947694778442383, "learning_rate": 1.9791666666666665e-05, "loss": 0.0862, "step": 35 }, { "epoch": 2.0, "grad_norm": 0.9083338379859924, "learning_rate": 1.8750000000000002e-05, "loss": 0.0167, "step": 36 }, { "epoch": 2.0, "eval_accuracy": 0.8888888888888888, "eval_f1_macro": 0.8999999999999999, "eval_f1_micro": 0.8888888888888888, "eval_f1_weighted": 0.8888888888888888, "eval_loss": 0.2903565764427185, "eval_precision_macro": 0.9166666666666666, "eval_precision_micro": 0.8888888888888888, "eval_precision_weighted": 0.9259259259259258, "eval_recall_macro": 0.9166666666666666, "eval_recall_micro": 0.8888888888888888, "eval_recall_weighted": 0.8888888888888888, "eval_runtime": 9.5629, "eval_samples_per_second": 0.941, "eval_steps_per_second": 0.314, "step": 36 } ], "logging_steps": 1, "max_steps": 54, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 500, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 5, "early_stopping_threshold": 0.01 }, "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": 1.917552219512832e+16, "train_batch_size": 2, "trial_name": null, "trial_params": null }