| { | |
| "best_metric": 0.9020938348196975, | |
| "best_model_checkpoint": "tiny-bert-sst2-distilled/run-41/checkpoint-3870", | |
| "epoch": 7.0, | |
| "eval_steps": 500, | |
| "global_step": 4515, | |
| "is_hyper_param_search": true, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 1.0, | |
| "grad_norm": 1.1879228353500366, | |
| "learning_rate": 0.0001500473390748166, | |
| "loss": 0.31, | |
| "step": 645 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.8937572702597906, | |
| "eval_f1": 0.1189710610932476, | |
| "eval_loss": 0.288364440202713, | |
| "eval_mcc": 0.19117679563293483, | |
| "eval_precision": 0.6981132075471698, | |
| "eval_recall": 0.06502636203866433, | |
| "eval_runtime": 9.9344, | |
| "eval_samples_per_second": 519.205, | |
| "eval_steps_per_second": 16.307, | |
| "step": 645 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 0.5299654006958008, | |
| "learning_rate": 0.00012861200492127138, | |
| "loss": 0.2852, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.8978286157425359, | |
| "eval_f1": 0.20030349013657056, | |
| "eval_loss": 0.2860374450683594, | |
| "eval_mcc": 0.2650150871029742, | |
| "eval_precision": 0.7333333333333333, | |
| "eval_recall": 0.11599297012302284, | |
| "eval_runtime": 9.9216, | |
| "eval_samples_per_second": 519.878, | |
| "eval_steps_per_second": 16.328, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "grad_norm": 0.777714729309082, | |
| "learning_rate": 0.00010717667076772616, | |
| "loss": 0.2784, | |
| "step": 1935 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.8984102365257852, | |
| "eval_f1": 0.19384615384615386, | |
| "eval_loss": 0.27609962224960327, | |
| "eval_mcc": 0.26911180968873705, | |
| "eval_precision": 0.7777777777777778, | |
| "eval_recall": 0.11072056239015818, | |
| "eval_runtime": 9.9163, | |
| "eval_samples_per_second": 520.153, | |
| "eval_steps_per_second": 16.337, | |
| "step": 1935 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "grad_norm": 1.847649097442627, | |
| "learning_rate": 8.574133661418092e-05, | |
| "loss": 0.2737, | |
| "step": 2580 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.9001550988755331, | |
| "eval_f1": 0.31057563587684067, | |
| "eval_loss": 0.27698394656181335, | |
| "eval_mcc": 0.3267057277830052, | |
| "eval_precision": 0.651685393258427, | |
| "eval_recall": 0.2038664323374341, | |
| "eval_runtime": 9.9379, | |
| "eval_samples_per_second": 519.024, | |
| "eval_steps_per_second": 16.301, | |
| "step": 2580 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "grad_norm": 1.6004267930984497, | |
| "learning_rate": 6.430600246063569e-05, | |
| "loss": 0.2705, | |
| "step": 3225 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.9009305932531989, | |
| "eval_f1": 0.328515111695138, | |
| "eval_loss": 0.2716519832611084, | |
| "eval_mcc": 0.33938446824512, | |
| "eval_precision": 0.6510416666666666, | |
| "eval_recall": 0.21968365553602812, | |
| "eval_runtime": 9.9296, | |
| "eval_samples_per_second": 519.457, | |
| "eval_steps_per_second": 16.315, | |
| "step": 3225 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "grad_norm": 1.280684232711792, | |
| "learning_rate": 4.287066830709046e-05, | |
| "loss": 0.2677, | |
| "step": 3870 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.9020938348196975, | |
| "eval_f1": 0.2995839112343967, | |
| "eval_loss": 0.268646240234375, | |
| "eval_mcc": 0.3338469363555913, | |
| "eval_precision": 0.7105263157894737, | |
| "eval_recall": 0.18980667838312829, | |
| "eval_runtime": 9.917, | |
| "eval_samples_per_second": 520.117, | |
| "eval_steps_per_second": 16.336, | |
| "step": 3870 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "grad_norm": 0.7534862160682678, | |
| "learning_rate": 2.143533415354523e-05, | |
| "loss": 0.2652, | |
| "step": 4515 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.9015122140364482, | |
| "eval_f1": 0.36499999999999994, | |
| "eval_loss": 0.2695741355419159, | |
| "eval_mcc": 0.360594280267097, | |
| "eval_precision": 0.6320346320346321, | |
| "eval_recall": 0.2565905096660808, | |
| "eval_runtime": 9.9457, | |
| "eval_samples_per_second": 518.615, | |
| "eval_steps_per_second": 16.288, | |
| "step": 4515 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 5160, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 8, | |
| "save_steps": 500, | |
| "total_flos": 11108597728200.0, | |
| "train_batch_size": 32, | |
| "trial_name": null, | |
| "trial_params": { | |
| "alpha": 0.886564686264361, | |
| "learning_rate": 0.00017148267322836185, | |
| "num_train_epochs": 8, | |
| "temperature": 15 | |
| } | |
| } | |