|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 3.0, |
|
"eval_steps": 500, |
|
"global_step": 1197, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 0.96484375, |
|
"learning_rate": 1.1111111111111113e-05, |
|
"loss": 0.4919, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 0.68359375, |
|
"learning_rate": 1.9999414238190722e-05, |
|
"loss": 0.3018, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 0.59765625, |
|
"learning_rate": 1.9978919779420427e-05, |
|
"loss": 0.2632, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.9929205817437334e-05, |
|
"loss": 0.2528, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 1.9850417920830427e-05, |
|
"loss": 0.2423, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"grad_norm": 0.58203125, |
|
"learning_rate": 1.9742786790241548e-05, |
|
"loss": 0.2366, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"grad_norm": 0.53515625, |
|
"learning_rate": 1.9606627582845587e-05, |
|
"loss": 0.2325, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"grad_norm": 0.54296875, |
|
"learning_rate": 1.944233898953152e-05, |
|
"loss": 0.2294, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.9250402067486523e-05, |
|
"loss": 0.2258, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 1.9031378831601425e-05, |
|
"loss": 0.2233, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 1.878591060882203e-05, |
|
"loss": 0.2191, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"grad_norm": 0.56640625, |
|
"learning_rate": 1.851471616026505e-05, |
|
"loss": 0.2178, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.8218589576597203e-05, |
|
"loss": 0.2141, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 1.789839795284022e-05, |
|
"loss": 0.2169, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"grad_norm": 0.546875, |
|
"learning_rate": 1.7555078849410044e-05, |
|
"loss": 0.2144, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"grad_norm": 0.81640625, |
|
"learning_rate": 1.7189637546824752e-05, |
|
"loss": 0.2083, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.680314410211967e-05, |
|
"loss": 0.2087, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"grad_norm": 0.56640625, |
|
"learning_rate": 1.6396730215588913e-05, |
|
"loss": 0.2089, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"grad_norm": 0.5390625, |
|
"learning_rate": 1.5971585917027864e-05, |
|
"loss": 0.2073, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.552895608117971e-05, |
|
"loss": 0.2058, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 1.05, |
|
"grad_norm": 0.5703125, |
|
"learning_rate": 1.5070136782589236e-05, |
|
"loss": 0.1761, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"grad_norm": 0.58984375, |
|
"learning_rate": 1.4596471500537334e-05, |
|
"loss": 0.1746, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 1.15, |
|
"grad_norm": 0.56640625, |
|
"learning_rate": 1.4109347185168577e-05, |
|
"loss": 0.175, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"grad_norm": 0.6015625, |
|
"learning_rate": 1.3610190196330777e-05, |
|
"loss": 0.1722, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"grad_norm": 0.62109375, |
|
"learning_rate": 1.3100462127018108e-05, |
|
"loss": 0.1749, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 1.3, |
|
"grad_norm": 0.58984375, |
|
"learning_rate": 1.2581655523647234e-05, |
|
"loss": 0.1742, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 1.35, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 1.2055289515698008e-05, |
|
"loss": 0.1729, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 1.1522905367515705e-05, |
|
"loss": 0.1733, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"grad_norm": 0.77734375, |
|
"learning_rate": 1.0986061965299646e-05, |
|
"loss": 0.1711, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 1.5, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 1.0446331252492864e-05, |
|
"loss": 0.1741, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 1.55, |
|
"grad_norm": 0.5625, |
|
"learning_rate": 9.905293626938564e-06, |
|
"loss": 0.1712, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"grad_norm": 0.60546875, |
|
"learning_rate": 9.364533313281056e-06, |
|
"loss": 0.1699, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"grad_norm": 0.609375, |
|
"learning_rate": 8.825633724161336e-06, |
|
"loss": 0.1714, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 1.7, |
|
"grad_norm": 0.57421875, |
|
"learning_rate": 8.290172823790337e-06, |
|
"loss": 0.1689, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 1.75, |
|
"grad_norm": 0.6015625, |
|
"learning_rate": 7.759718507475822e-06, |
|
"loss": 0.1703, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 1.8, |
|
"grad_norm": 0.578125, |
|
"learning_rate": 7.235824010632284e-06, |
|
"loss": 0.1708, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"grad_norm": 0.59765625, |
|
"learning_rate": 6.720023360716818e-06, |
|
"loss": 0.171, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 1.9, |
|
"grad_norm": 0.60546875, |
|
"learning_rate": 6.2138268854082094e-06, |
|
"loss": 0.1667, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 1.95, |
|
"grad_norm": 0.6015625, |
|
"learning_rate": 5.7187167901819665e-06, |
|
"loss": 0.1709, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 2.01, |
|
"grad_norm": 0.56640625, |
|
"learning_rate": 5.236142818230518e-06, |
|
"loss": 0.1676, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 2.06, |
|
"grad_norm": 0.59765625, |
|
"learning_rate": 4.767518005437073e-06, |
|
"loss": 0.1536, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 2.11, |
|
"grad_norm": 0.609375, |
|
"learning_rate": 4.314214542832889e-06, |
|
"loss": 0.1526, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 2.16, |
|
"grad_norm": 0.59375, |
|
"learning_rate": 3.877559758653358e-06, |
|
"loss": 0.1559, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 2.21, |
|
"grad_norm": 0.63671875, |
|
"learning_rate": 3.4588322317578395e-06, |
|
"loss": 0.1529, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"grad_norm": 0.5703125, |
|
"learning_rate": 3.0592580477936606e-06, |
|
"loss": 0.1515, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 2.31, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 2.6800072090667085e-06, |
|
"loss": 0.1516, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 2.36, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 2.3221902086308943e-06, |
|
"loss": 0.1515, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 2.41, |
|
"grad_norm": 0.6015625, |
|
"learning_rate": 1.986854778628067e-06, |
|
"loss": 0.1533, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 2.46, |
|
"grad_norm": 0.64453125, |
|
"learning_rate": 1.67498282239956e-06, |
|
"loss": 0.1528, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 2.51, |
|
"grad_norm": 0.59375, |
|
"learning_rate": 1.3874875393525722e-06, |
|
"loss": 0.1508, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 2.56, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 1.1252107510000843e-06, |
|
"loss": 0.1519, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 2.61, |
|
"grad_norm": 0.625, |
|
"learning_rate": 8.889204360040471e-07, |
|
"loss": 0.1502, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 2.66, |
|
"grad_norm": 0.62890625, |
|
"learning_rate": 6.793084814394945e-07, |
|
"loss": 0.1526, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 2.71, |
|
"grad_norm": 0.62109375, |
|
"learning_rate": 4.969886568641757e-07, |
|
"loss": 0.1513, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 2.76, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 3.424948171258824e-07, |
|
"loss": 0.1525, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 2.81, |
|
"grad_norm": 0.62890625, |
|
"learning_rate": 2.16279339169847e-07, |
|
"loss": 0.1549, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 2.86, |
|
"grad_norm": 0.6171875, |
|
"learning_rate": 1.1871179742348416e-07, |
|
"loss": 0.1494, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"grad_norm": 0.61328125, |
|
"learning_rate": 5.007788163703842e-08, |
|
"loss": 0.1536, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 2.96, |
|
"grad_norm": 0.65234375, |
|
"learning_rate": 1.0578560348905031e-08, |
|
"loss": 0.1532, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"step": 1197, |
|
"total_flos": 1.7005983975572767e+18, |
|
"train_loss": 0.18855523903766272, |
|
"train_runtime": 5513.3521, |
|
"train_samples_per_second": 31.264, |
|
"train_steps_per_second": 0.217 |
|
} |
|
], |
|
"logging_steps": 20, |
|
"max_steps": 1197, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 10000, |
|
"total_flos": 1.7005983975572767e+18, |
|
"train_batch_size": 36, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|