| { | |
| "best_metric": 3.7167649269104004, | |
| "best_model_checkpoint": "models/GPT2_natural_function_67/checkpoint-64220", | |
| "epoch": 10.0, | |
| "global_step": 64220, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 1e-05, | |
| "loss": 7.6403, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 2e-05, | |
| "loss": 6.1944, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 3e-05, | |
| "loss": 5.7388, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 4e-05, | |
| "loss": 5.4181, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 5e-05, | |
| "loss": 5.1669, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "learning_rate": 6e-05, | |
| "loss": 4.97, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.28050345574375357, | |
| "eval_loss": 4.745330333709717, | |
| "eval_runtime": 2.2437, | |
| "eval_samples_per_second": 525.032, | |
| "eval_steps_per_second": 4.457, | |
| "step": 6422 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "learning_rate": 7e-05, | |
| "loss": 4.7993, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 8e-05, | |
| "loss": 4.6673, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 1.4, | |
| "learning_rate": 9e-05, | |
| "loss": 4.5629, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 1.56, | |
| "learning_rate": 0.0001, | |
| "loss": 4.4737, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 1.71, | |
| "learning_rate": 9.815750645518259e-05, | |
| "loss": 4.399, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 1.87, | |
| "learning_rate": 9.631316857248248e-05, | |
| "loss": 4.3346, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.31933879657233, | |
| "eval_loss": 4.193398952484131, | |
| "eval_runtime": 2.2441, | |
| "eval_samples_per_second": 524.934, | |
| "eval_steps_per_second": 4.456, | |
| "step": 12844 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "learning_rate": 9.447067502766507e-05, | |
| "loss": 4.2713, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 2.18, | |
| "learning_rate": 9.262633714496496e-05, | |
| "loss": 4.1959, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 2.34, | |
| "learning_rate": 9.078384360014756e-05, | |
| "loss": 4.1675, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 2.49, | |
| "learning_rate": 8.893950571744743e-05, | |
| "loss": 4.1374, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 2.65, | |
| "learning_rate": 8.709701217263003e-05, | |
| "loss": 4.1132, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 2.8, | |
| "learning_rate": 8.525267428992992e-05, | |
| "loss": 4.0906, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 2.96, | |
| "learning_rate": 8.341018074511251e-05, | |
| "loss": 4.0708, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.33757335935724503, | |
| "eval_loss": 3.999851703643799, | |
| "eval_runtime": 2.2451, | |
| "eval_samples_per_second": 524.706, | |
| "eval_steps_per_second": 4.454, | |
| "step": 19266 | |
| }, | |
| { | |
| "epoch": 3.11, | |
| "learning_rate": 8.15658428624124e-05, | |
| "loss": 4.0029, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 3.27, | |
| "learning_rate": 7.972334931759499e-05, | |
| "loss": 3.9806, | |
| "step": 21000 | |
| }, | |
| { | |
| "epoch": 3.43, | |
| "learning_rate": 7.787901143489488e-05, | |
| "loss": 3.971, | |
| "step": 22000 | |
| }, | |
| { | |
| "epoch": 3.58, | |
| "learning_rate": 7.603651789007746e-05, | |
| "loss": 3.961, | |
| "step": 23000 | |
| }, | |
| { | |
| "epoch": 3.74, | |
| "learning_rate": 7.419218000737735e-05, | |
| "loss": 3.9509, | |
| "step": 24000 | |
| }, | |
| { | |
| "epoch": 3.89, | |
| "learning_rate": 7.234968646255994e-05, | |
| "loss": 3.9406, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.34709837840728314, | |
| "eval_loss": 3.901829719543457, | |
| "eval_runtime": 2.2445, | |
| "eval_samples_per_second": 524.829, | |
| "eval_steps_per_second": 4.455, | |
| "step": 25688 | |
| }, | |
| { | |
| "epoch": 4.05, | |
| "learning_rate": 7.050534857985983e-05, | |
| "loss": 3.9065, | |
| "step": 26000 | |
| }, | |
| { | |
| "epoch": 4.2, | |
| "learning_rate": 6.866469937292512e-05, | |
| "loss": 3.8622, | |
| "step": 27000 | |
| }, | |
| { | |
| "epoch": 4.36, | |
| "learning_rate": 6.682036149022502e-05, | |
| "loss": 3.8579, | |
| "step": 28000 | |
| }, | |
| { | |
| "epoch": 4.52, | |
| "learning_rate": 6.497602360752491e-05, | |
| "loss": 3.8586, | |
| "step": 29000 | |
| }, | |
| { | |
| "epoch": 4.67, | |
| "learning_rate": 6.31316857248248e-05, | |
| "loss": 3.8531, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 4.83, | |
| "learning_rate": 6.128919218000738e-05, | |
| "loss": 3.8469, | |
| "step": 31000 | |
| }, | |
| { | |
| "epoch": 4.98, | |
| "learning_rate": 5.9444854297307275e-05, | |
| "loss": 3.8415, | |
| "step": 32000 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.35395639212331054, | |
| "eval_loss": 3.840256690979004, | |
| "eval_runtime": 2.247, | |
| "eval_samples_per_second": 524.252, | |
| "eval_steps_per_second": 4.45, | |
| "step": 32110 | |
| }, | |
| { | |
| "epoch": 5.14, | |
| "learning_rate": 5.760236075248986e-05, | |
| "loss": 3.7757, | |
| "step": 33000 | |
| }, | |
| { | |
| "epoch": 5.29, | |
| "learning_rate": 5.5759867207672446e-05, | |
| "loss": 3.776, | |
| "step": 34000 | |
| }, | |
| { | |
| "epoch": 5.45, | |
| "learning_rate": 5.391552932497234e-05, | |
| "loss": 3.7774, | |
| "step": 35000 | |
| }, | |
| { | |
| "epoch": 5.61, | |
| "learning_rate": 5.207119144227223e-05, | |
| "loss": 3.7758, | |
| "step": 36000 | |
| }, | |
| { | |
| "epoch": 5.76, | |
| "learning_rate": 5.022685355957212e-05, | |
| "loss": 3.7736, | |
| "step": 37000 | |
| }, | |
| { | |
| "epoch": 5.92, | |
| "learning_rate": 4.8384360014754705e-05, | |
| "loss": 3.7713, | |
| "step": 38000 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.35874229643196126, | |
| "eval_loss": 3.797034978866577, | |
| "eval_runtime": 2.2428, | |
| "eval_samples_per_second": 525.234, | |
| "eval_steps_per_second": 4.459, | |
| "step": 38532 | |
| }, | |
| { | |
| "epoch": 6.07, | |
| "learning_rate": 4.6540022132054597e-05, | |
| "loss": 3.734, | |
| "step": 39000 | |
| }, | |
| { | |
| "epoch": 6.23, | |
| "learning_rate": 4.469752858723719e-05, | |
| "loss": 3.7046, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 6.38, | |
| "learning_rate": 4.285319070453707e-05, | |
| "loss": 3.7108, | |
| "step": 41000 | |
| }, | |
| { | |
| "epoch": 6.54, | |
| "learning_rate": 4.1010697159719665e-05, | |
| "loss": 3.7108, | |
| "step": 42000 | |
| }, | |
| { | |
| "epoch": 6.7, | |
| "learning_rate": 3.916635927701956e-05, | |
| "loss": 3.7112, | |
| "step": 43000 | |
| }, | |
| { | |
| "epoch": 6.85, | |
| "learning_rate": 3.732202139431944e-05, | |
| "loss": 3.7109, | |
| "step": 44000 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.3625523040519765, | |
| "eval_loss": 3.76499080657959, | |
| "eval_runtime": 2.2423, | |
| "eval_samples_per_second": 525.348, | |
| "eval_steps_per_second": 4.46, | |
| "step": 44954 | |
| }, | |
| { | |
| "epoch": 7.01, | |
| "learning_rate": 3.547768351161933e-05, | |
| "loss": 3.7065, | |
| "step": 45000 | |
| }, | |
| { | |
| "epoch": 7.16, | |
| "learning_rate": 3.3635189966801925e-05, | |
| "loss": 3.646, | |
| "step": 46000 | |
| }, | |
| { | |
| "epoch": 7.32, | |
| "learning_rate": 3.179085208410181e-05, | |
| "loss": 3.6519, | |
| "step": 47000 | |
| }, | |
| { | |
| "epoch": 7.47, | |
| "learning_rate": 2.9948358539284398e-05, | |
| "loss": 3.6559, | |
| "step": 48000 | |
| }, | |
| { | |
| "epoch": 7.63, | |
| "learning_rate": 2.810402065658429e-05, | |
| "loss": 3.6561, | |
| "step": 49000 | |
| }, | |
| { | |
| "epoch": 7.79, | |
| "learning_rate": 2.6259682773884174e-05, | |
| "loss": 3.6566, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 7.94, | |
| "learning_rate": 2.4417189229066766e-05, | |
| "loss": 3.6557, | |
| "step": 51000 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.3650722564603024, | |
| "eval_loss": 3.740530252456665, | |
| "eval_runtime": 2.2409, | |
| "eval_samples_per_second": 525.684, | |
| "eval_steps_per_second": 4.463, | |
| "step": 51376 | |
| }, | |
| { | |
| "epoch": 8.1, | |
| "learning_rate": 2.2572851346366654e-05, | |
| "loss": 3.6199, | |
| "step": 52000 | |
| }, | |
| { | |
| "epoch": 8.25, | |
| "learning_rate": 2.0730357801549243e-05, | |
| "loss": 3.6032, | |
| "step": 53000 | |
| }, | |
| { | |
| "epoch": 8.41, | |
| "learning_rate": 1.8886019918849134e-05, | |
| "loss": 3.6066, | |
| "step": 54000 | |
| }, | |
| { | |
| "epoch": 8.56, | |
| "learning_rate": 1.7043526374031723e-05, | |
| "loss": 3.6088, | |
| "step": 55000 | |
| }, | |
| { | |
| "epoch": 8.72, | |
| "learning_rate": 1.5199188491331613e-05, | |
| "loss": 3.6074, | |
| "step": 56000 | |
| }, | |
| { | |
| "epoch": 8.88, | |
| "learning_rate": 1.3356694946514203e-05, | |
| "loss": 3.6097, | |
| "step": 57000 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.36772589334652356, | |
| "eval_loss": 3.724553108215332, | |
| "eval_runtime": 2.2415, | |
| "eval_samples_per_second": 525.553, | |
| "eval_steps_per_second": 4.461, | |
| "step": 57798 | |
| }, | |
| { | |
| "epoch": 9.03, | |
| "learning_rate": 1.1512357063814091e-05, | |
| "loss": 3.5989, | |
| "step": 58000 | |
| }, | |
| { | |
| "epoch": 9.19, | |
| "learning_rate": 9.66986351899668e-06, | |
| "loss": 3.5648, | |
| "step": 59000 | |
| }, | |
| { | |
| "epoch": 9.34, | |
| "learning_rate": 7.82552563629657e-06, | |
| "loss": 3.5665, | |
| "step": 60000 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "learning_rate": 5.9830320914791595e-06, | |
| "loss": 3.568, | |
| "step": 61000 | |
| }, | |
| { | |
| "epoch": 9.65, | |
| "learning_rate": 4.138694208779048e-06, | |
| "loss": 3.5691, | |
| "step": 62000 | |
| }, | |
| { | |
| "epoch": 9.81, | |
| "learning_rate": 2.296200663961638e-06, | |
| "loss": 3.5672, | |
| "step": 63000 | |
| }, | |
| { | |
| "epoch": 9.97, | |
| "learning_rate": 4.5186278126152714e-07, | |
| "loss": 3.5655, | |
| "step": 64000 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.36849457909442135, | |
| "eval_loss": 3.7167649269104004, | |
| "eval_runtime": 2.2439, | |
| "eval_samples_per_second": 524.989, | |
| "eval_steps_per_second": 4.457, | |
| "step": 64220 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 64220, | |
| "total_flos": 5.3695120637952e+17, | |
| "train_loss": 4.052415334677184, | |
| "train_runtime": 32698.3408, | |
| "train_samples_per_second": 251.387, | |
| "train_steps_per_second": 1.964 | |
| } | |
| ], | |
| "max_steps": 64220, | |
| "num_train_epochs": 10, | |
| "total_flos": 5.3695120637952e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |