| { | |
| "best_metric": 0.14040808379650116, | |
| "best_model_checkpoint": "./results/checkpoint-51064", | |
| "epoch": 2.0, | |
| "eval_steps": 500, | |
| "global_step": 51064, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 1.986944487962818e-05, | |
| "loss": 0.2496, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 1.9738889759256358e-05, | |
| "loss": 0.208, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 1.9608334638884537e-05, | |
| "loss": 0.1884, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 1.9477779518512717e-05, | |
| "loss": 0.1955, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 1.9347224398140897e-05, | |
| "loss": 0.2005, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 1.9216669277769077e-05, | |
| "loss": 0.1625, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 1.9086114157397256e-05, | |
| "loss": 0.18, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 1.8955559037025433e-05, | |
| "loss": 0.153, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 1.8825003916653613e-05, | |
| "loss": 0.1996, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 1.8694448796281792e-05, | |
| "loss": 0.1643, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 1.856389367590997e-05, | |
| "loss": 0.2022, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 1.843333855553815e-05, | |
| "loss": 0.1706, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 1.8302783435166328e-05, | |
| "loss": 0.1895, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 1.8172228314794508e-05, | |
| "loss": 0.1733, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 1.8041673194422688e-05, | |
| "loss": 0.1938, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 1.7911118074050867e-05, | |
| "loss": 0.1786, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 1.7780562953679047e-05, | |
| "loss": 0.19, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 1.7650007833307223e-05, | |
| "loss": 0.1908, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 1.7519452712935403e-05, | |
| "loss": 0.1755, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 1.7388897592563583e-05, | |
| "loss": 0.1536, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "learning_rate": 1.725834247219176e-05, | |
| "loss": 0.1574, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "learning_rate": 1.712778735181994e-05, | |
| "loss": 0.1472, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "learning_rate": 1.699723223144812e-05, | |
| "loss": 0.1936, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 1.68666771110763e-05, | |
| "loss": 0.1746, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "learning_rate": 1.6736121990704478e-05, | |
| "loss": 0.1731, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 1.6605566870332658e-05, | |
| "loss": 0.1927, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "learning_rate": 1.6475011749960834e-05, | |
| "loss": 0.1946, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 1.6344456629589014e-05, | |
| "loss": 0.1636, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 1.6213901509217194e-05, | |
| "loss": 0.2385, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 1.608334638884537e-05, | |
| "loss": 0.2137, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 1.595279126847355e-05, | |
| "loss": 0.1736, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 1.582223614810173e-05, | |
| "loss": 0.2196, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 1.569168102772991e-05, | |
| "loss": 0.1878, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 1.556112590735809e-05, | |
| "loss": 0.1742, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "learning_rate": 1.543057078698627e-05, | |
| "loss": 0.1907, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 1.5300015666614445e-05, | |
| "loss": 0.2048, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "learning_rate": 1.5169460546242625e-05, | |
| "loss": 0.1777, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "learning_rate": 1.5038905425870805e-05, | |
| "loss": 0.213, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "learning_rate": 1.4908350305498983e-05, | |
| "loss": 0.1941, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 1.4777795185127163e-05, | |
| "loss": 0.1525, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 1.4647240064755342e-05, | |
| "loss": 0.1638, | |
| "step": 20500 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "learning_rate": 1.4516684944383519e-05, | |
| "loss": 0.174, | |
| "step": 21000 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "learning_rate": 1.4386129824011698e-05, | |
| "loss": 0.1693, | |
| "step": 21500 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 1.4255574703639878e-05, | |
| "loss": 0.1926, | |
| "step": 22000 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 1.4125019583268056e-05, | |
| "loss": 0.1915, | |
| "step": 22500 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.3994464462896236e-05, | |
| "loss": 0.1671, | |
| "step": 23000 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "learning_rate": 1.3863909342524416e-05, | |
| "loss": 0.1572, | |
| "step": 23500 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 1.3733354222152594e-05, | |
| "loss": 0.1613, | |
| "step": 24000 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 1.3602799101780773e-05, | |
| "loss": 0.165, | |
| "step": 24500 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "learning_rate": 1.3472243981408951e-05, | |
| "loss": 0.1739, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 1.3341688861037131e-05, | |
| "loss": 0.1611, | |
| "step": 25500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_loss": 0.17089270055294037, | |
| "eval_runtime": 899.1592, | |
| "eval_samples_per_second": 35.494, | |
| "eval_steps_per_second": 7.099, | |
| "step": 25532 | |
| }, | |
| { | |
| "epoch": 1.02, | |
| "learning_rate": 1.321113374066531e-05, | |
| "loss": 0.1776, | |
| "step": 26000 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "learning_rate": 1.3080578620293489e-05, | |
| "loss": 0.1759, | |
| "step": 26500 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "learning_rate": 1.2950023499921669e-05, | |
| "loss": 0.1544, | |
| "step": 27000 | |
| }, | |
| { | |
| "epoch": 1.08, | |
| "learning_rate": 1.2819468379549847e-05, | |
| "loss": 0.1691, | |
| "step": 27500 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "learning_rate": 1.2688913259178027e-05, | |
| "loss": 0.1866, | |
| "step": 28000 | |
| }, | |
| { | |
| "epoch": 1.12, | |
| "learning_rate": 1.2558358138806206e-05, | |
| "loss": 0.1833, | |
| "step": 28500 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "learning_rate": 1.2427803018434383e-05, | |
| "loss": 0.1791, | |
| "step": 29000 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "learning_rate": 1.2297247898062562e-05, | |
| "loss": 0.173, | |
| "step": 29500 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "learning_rate": 1.2166692777690742e-05, | |
| "loss": 0.2536, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "learning_rate": 1.203613765731892e-05, | |
| "loss": 0.2705, | |
| "step": 30500 | |
| }, | |
| { | |
| "epoch": 1.21, | |
| "learning_rate": 1.19055825369471e-05, | |
| "loss": 0.2602, | |
| "step": 31000 | |
| }, | |
| { | |
| "epoch": 1.23, | |
| "learning_rate": 1.177502741657528e-05, | |
| "loss": 0.1628, | |
| "step": 31500 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 1.1644472296203458e-05, | |
| "loss": 0.249, | |
| "step": 32000 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "learning_rate": 1.1513917175831637e-05, | |
| "loss": 0.2991, | |
| "step": 32500 | |
| }, | |
| { | |
| "epoch": 1.29, | |
| "learning_rate": 1.1383362055459817e-05, | |
| "loss": 0.3618, | |
| "step": 33000 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "learning_rate": 1.1252806935087994e-05, | |
| "loss": 0.3487, | |
| "step": 33500 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 1.1122251814716173e-05, | |
| "loss": 0.3179, | |
| "step": 34000 | |
| }, | |
| { | |
| "epoch": 1.35, | |
| "learning_rate": 1.0991696694344353e-05, | |
| "loss": 0.3457, | |
| "step": 34500 | |
| }, | |
| { | |
| "epoch": 1.37, | |
| "learning_rate": 1.0861141573972531e-05, | |
| "loss": 0.3574, | |
| "step": 35000 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "learning_rate": 1.073058645360071e-05, | |
| "loss": 0.3486, | |
| "step": 35500 | |
| }, | |
| { | |
| "epoch": 1.41, | |
| "learning_rate": 1.060003133322889e-05, | |
| "loss": 0.2576, | |
| "step": 36000 | |
| }, | |
| { | |
| "epoch": 1.43, | |
| "learning_rate": 1.0469476212857069e-05, | |
| "loss": 0.1753, | |
| "step": 36500 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "learning_rate": 1.0338921092485248e-05, | |
| "loss": 0.2012, | |
| "step": 37000 | |
| }, | |
| { | |
| "epoch": 1.47, | |
| "learning_rate": 1.0208365972113428e-05, | |
| "loss": 0.2095, | |
| "step": 37500 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "learning_rate": 1.0077810851741604e-05, | |
| "loss": 0.1966, | |
| "step": 38000 | |
| }, | |
| { | |
| "epoch": 1.51, | |
| "learning_rate": 9.947255731369784e-06, | |
| "loss": 0.1492, | |
| "step": 38500 | |
| }, | |
| { | |
| "epoch": 1.53, | |
| "learning_rate": 9.816700610997964e-06, | |
| "loss": 0.1663, | |
| "step": 39000 | |
| }, | |
| { | |
| "epoch": 1.55, | |
| "learning_rate": 9.686145490626144e-06, | |
| "loss": 0.158, | |
| "step": 39500 | |
| }, | |
| { | |
| "epoch": 1.57, | |
| "learning_rate": 9.555590370254322e-06, | |
| "loss": 0.1635, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 1.59, | |
| "learning_rate": 9.425035249882501e-06, | |
| "loss": 0.1605, | |
| "step": 40500 | |
| }, | |
| { | |
| "epoch": 1.61, | |
| "learning_rate": 9.29448012951068e-06, | |
| "loss": 0.1479, | |
| "step": 41000 | |
| }, | |
| { | |
| "epoch": 1.63, | |
| "learning_rate": 9.16392500913886e-06, | |
| "loss": 0.1555, | |
| "step": 41500 | |
| }, | |
| { | |
| "epoch": 1.64, | |
| "learning_rate": 9.033369888767039e-06, | |
| "loss": 0.1657, | |
| "step": 42000 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "learning_rate": 8.902814768395217e-06, | |
| "loss": 0.1322, | |
| "step": 42500 | |
| }, | |
| { | |
| "epoch": 1.68, | |
| "learning_rate": 8.772259648023395e-06, | |
| "loss": 0.1441, | |
| "step": 43000 | |
| }, | |
| { | |
| "epoch": 1.7, | |
| "learning_rate": 8.641704527651575e-06, | |
| "loss": 0.1514, | |
| "step": 43500 | |
| }, | |
| { | |
| "epoch": 1.72, | |
| "learning_rate": 8.511149407279755e-06, | |
| "loss": 0.1556, | |
| "step": 44000 | |
| }, | |
| { | |
| "epoch": 1.74, | |
| "learning_rate": 8.380594286907934e-06, | |
| "loss": 0.2028, | |
| "step": 44500 | |
| }, | |
| { | |
| "epoch": 1.76, | |
| "learning_rate": 8.250039166536112e-06, | |
| "loss": 0.1538, | |
| "step": 45000 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "learning_rate": 8.11948404616429e-06, | |
| "loss": 0.1804, | |
| "step": 45500 | |
| }, | |
| { | |
| "epoch": 1.8, | |
| "learning_rate": 7.98892892579247e-06, | |
| "loss": 0.146, | |
| "step": 46000 | |
| }, | |
| { | |
| "epoch": 1.82, | |
| "learning_rate": 7.85837380542065e-06, | |
| "loss": 0.1538, | |
| "step": 46500 | |
| }, | |
| { | |
| "epoch": 1.84, | |
| "learning_rate": 7.727818685048828e-06, | |
| "loss": 0.1476, | |
| "step": 47000 | |
| }, | |
| { | |
| "epoch": 1.86, | |
| "learning_rate": 7.597263564677008e-06, | |
| "loss": 0.1766, | |
| "step": 47500 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "learning_rate": 7.4667084443051865e-06, | |
| "loss": 0.1807, | |
| "step": 48000 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 7.336153323933365e-06, | |
| "loss": 0.1416, | |
| "step": 48500 | |
| }, | |
| { | |
| "epoch": 1.92, | |
| "learning_rate": 7.205598203561544e-06, | |
| "loss": 0.1453, | |
| "step": 49000 | |
| }, | |
| { | |
| "epoch": 1.94, | |
| "learning_rate": 7.075043083189723e-06, | |
| "loss": 0.1349, | |
| "step": 49500 | |
| }, | |
| { | |
| "epoch": 1.96, | |
| "learning_rate": 6.944487962817902e-06, | |
| "loss": 0.1852, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 1.98, | |
| "learning_rate": 6.813932842446082e-06, | |
| "loss": 0.1599, | |
| "step": 50500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 6.68337772207426e-06, | |
| "loss": 0.1184, | |
| "step": 51000 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_loss": 0.14040808379650116, | |
| "eval_runtime": 896.823, | |
| "eval_samples_per_second": 35.587, | |
| "eval_steps_per_second": 7.117, | |
| "step": 51064 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 76596, | |
| "num_train_epochs": 3, | |
| "save_steps": 500, | |
| "total_flos": 6.717540976607232e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |