| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.0, | |
| "eval_steps": 100, | |
| "global_step": 248, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.04032258064516129, | |
| "grad_norm": 1.870942234992981, | |
| "learning_rate": 1.967741935483871e-05, | |
| "loss": 1.3889, | |
| "num_tokens": 583069.0, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.08064516129032258, | |
| "grad_norm": 1.5309628248214722, | |
| "learning_rate": 1.9274193548387097e-05, | |
| "loss": 1.2548, | |
| "num_tokens": 1166377.0, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.12096774193548387, | |
| "grad_norm": 1.5700175762176514, | |
| "learning_rate": 1.8870967741935487e-05, | |
| "loss": 1.2989, | |
| "num_tokens": 1764845.0, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.16129032258064516, | |
| "grad_norm": 1.704406499862671, | |
| "learning_rate": 1.8467741935483873e-05, | |
| "loss": 1.2885, | |
| "num_tokens": 2359985.0, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.20161290322580644, | |
| "grad_norm": 1.4732401371002197, | |
| "learning_rate": 1.806451612903226e-05, | |
| "loss": 1.2504, | |
| "num_tokens": 2938303.0, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.24193548387096775, | |
| "grad_norm": 1.581865668296814, | |
| "learning_rate": 1.7661290322580645e-05, | |
| "loss": 1.2158, | |
| "num_tokens": 3536824.0, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.28225806451612906, | |
| "grad_norm": 1.515305757522583, | |
| "learning_rate": 1.7258064516129035e-05, | |
| "loss": 1.1587, | |
| "num_tokens": 4106428.0, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.3225806451612903, | |
| "grad_norm": 1.2818363904953003, | |
| "learning_rate": 1.685483870967742e-05, | |
| "loss": 1.228, | |
| "num_tokens": 4713410.0, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.3629032258064516, | |
| "grad_norm": 1.077473521232605, | |
| "learning_rate": 1.6451612903225807e-05, | |
| "loss": 1.2081, | |
| "num_tokens": 5302842.0, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.4032258064516129, | |
| "grad_norm": 0.8858934044837952, | |
| "learning_rate": 1.6048387096774193e-05, | |
| "loss": 1.1533, | |
| "num_tokens": 5883277.0, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.4435483870967742, | |
| "grad_norm": 0.71666020154953, | |
| "learning_rate": 1.5645161290322583e-05, | |
| "loss": 1.1602, | |
| "num_tokens": 6457162.0, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.4838709677419355, | |
| "grad_norm": 0.6815721988677979, | |
| "learning_rate": 1.5241935483870969e-05, | |
| "loss": 1.1474, | |
| "num_tokens": 7013145.0, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.5241935483870968, | |
| "grad_norm": 0.7701214551925659, | |
| "learning_rate": 1.4838709677419357e-05, | |
| "loss": 1.1307, | |
| "num_tokens": 7621530.0, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.5645161290322581, | |
| "grad_norm": 0.6789644956588745, | |
| "learning_rate": 1.4435483870967743e-05, | |
| "loss": 1.1396, | |
| "num_tokens": 8194938.0, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.6048387096774194, | |
| "grad_norm": 0.6866022944450378, | |
| "learning_rate": 1.4032258064516131e-05, | |
| "loss": 1.1321, | |
| "num_tokens": 8800144.0, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.6451612903225806, | |
| "grad_norm": 0.7200160622596741, | |
| "learning_rate": 1.3629032258064517e-05, | |
| "loss": 1.1685, | |
| "num_tokens": 9408277.0, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.6854838709677419, | |
| "grad_norm": 0.6781009435653687, | |
| "learning_rate": 1.3225806451612903e-05, | |
| "loss": 1.1287, | |
| "num_tokens": 10000739.0, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.7258064516129032, | |
| "grad_norm": 0.6929251551628113, | |
| "learning_rate": 1.2822580645161291e-05, | |
| "loss": 1.1286, | |
| "num_tokens": 10586681.0, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.7661290322580645, | |
| "grad_norm": 0.6937847137451172, | |
| "learning_rate": 1.2419354838709679e-05, | |
| "loss": 1.1326, | |
| "num_tokens": 11185064.0, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.8064516129032258, | |
| "grad_norm": 0.6759666204452515, | |
| "learning_rate": 1.2016129032258067e-05, | |
| "loss": 1.1341, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.8064516129032258, | |
| "eval_loss": 1.1217821836471558, | |
| "eval_num_tokens": 11776617.0, | |
| "eval_runtime": 0.6973, | |
| "eval_samples_per_second": 286.828, | |
| "eval_steps_per_second": 5.737, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.8467741935483871, | |
| "grad_norm": 0.6206675171852112, | |
| "learning_rate": 1.1612903225806453e-05, | |
| "loss": 1.0987, | |
| "num_tokens": 12370086.0, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.8870967741935484, | |
| "grad_norm": 0.6287978887557983, | |
| "learning_rate": 1.1209677419354839e-05, | |
| "loss": 1.1592, | |
| "num_tokens": 12949436.0, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.9274193548387096, | |
| "grad_norm": 0.6936933994293213, | |
| "learning_rate": 1.0806451612903225e-05, | |
| "loss": 1.1478, | |
| "num_tokens": 13537898.0, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.967741935483871, | |
| "grad_norm": 0.6776860356330872, | |
| "learning_rate": 1.0403225806451613e-05, | |
| "loss": 1.1115, | |
| "num_tokens": 14126751.0, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.0080645161290323, | |
| "grad_norm": 0.7314103841781616, | |
| "learning_rate": 1e-05, | |
| "loss": 1.0969, | |
| "num_tokens": 14734884.0, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 1.0483870967741935, | |
| "grad_norm": 0.7181735038757324, | |
| "learning_rate": 9.596774193548389e-06, | |
| "loss": 1.0159, | |
| "num_tokens": 15308122.0, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.0887096774193548, | |
| "grad_norm": 0.6945134401321411, | |
| "learning_rate": 9.193548387096775e-06, | |
| "loss": 0.9845, | |
| "num_tokens": 15893652.0, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 1.129032258064516, | |
| "grad_norm": 0.6895014643669128, | |
| "learning_rate": 8.790322580645163e-06, | |
| "loss": 1.0348, | |
| "num_tokens": 16479176.0, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.1693548387096775, | |
| "grad_norm": 0.6193928122520447, | |
| "learning_rate": 8.387096774193549e-06, | |
| "loss": 0.9616, | |
| "num_tokens": 17068227.0, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 1.2096774193548387, | |
| "grad_norm": 0.6025323271751404, | |
| "learning_rate": 7.983870967741935e-06, | |
| "loss": 0.9885, | |
| "num_tokens": 17648475.0, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 0.6143034100532532, | |
| "learning_rate": 7.580645161290323e-06, | |
| "loss": 0.9836, | |
| "num_tokens": 18252289.0, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 1.2903225806451613, | |
| "grad_norm": 0.5971494913101196, | |
| "learning_rate": 7.177419354838711e-06, | |
| "loss": 0.966, | |
| "num_tokens": 18842881.0, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.3306451612903225, | |
| "grad_norm": 0.5852988958358765, | |
| "learning_rate": 6.774193548387097e-06, | |
| "loss": 1.0077, | |
| "num_tokens": 19423540.0, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 1.370967741935484, | |
| "grad_norm": 0.6165304183959961, | |
| "learning_rate": 6.370967741935485e-06, | |
| "loss": 1.0357, | |
| "num_tokens": 20017275.0, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.4112903225806452, | |
| "grad_norm": 0.6567261815071106, | |
| "learning_rate": 5.967741935483872e-06, | |
| "loss": 1.0142, | |
| "num_tokens": 20592466.0, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 1.4516129032258065, | |
| "grad_norm": 0.6339277625083923, | |
| "learning_rate": 5.564516129032258e-06, | |
| "loss": 1.0359, | |
| "num_tokens": 21170579.0, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.4919354838709677, | |
| "grad_norm": 0.6214042901992798, | |
| "learning_rate": 5.161290322580646e-06, | |
| "loss": 0.9822, | |
| "num_tokens": 21779118.0, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 1.532258064516129, | |
| "grad_norm": 0.6341310143470764, | |
| "learning_rate": 4.758064516129033e-06, | |
| "loss": 0.96, | |
| "num_tokens": 22337296.0, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.5725806451612905, | |
| "grad_norm": 0.636868417263031, | |
| "learning_rate": 4.35483870967742e-06, | |
| "loss": 1.0217, | |
| "num_tokens": 22933384.0, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 1.6129032258064515, | |
| "grad_norm": 0.6980040073394775, | |
| "learning_rate": 3.951612903225807e-06, | |
| "loss": 0.9955, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.6129032258064515, | |
| "eval_loss": 1.1184884309768677, | |
| "eval_num_tokens": 23494529.0, | |
| "eval_runtime": 0.6648, | |
| "eval_samples_per_second": 300.826, | |
| "eval_steps_per_second": 6.017, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.653225806451613, | |
| "grad_norm": 0.6227007508277893, | |
| "learning_rate": 3.548387096774194e-06, | |
| "loss": 1.0197, | |
| "num_tokens": 24073440.0, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 1.6935483870967742, | |
| "grad_norm": 0.6243858337402344, | |
| "learning_rate": 3.145161290322581e-06, | |
| "loss": 1.0462, | |
| "num_tokens": 24679112.0, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.7338709677419355, | |
| "grad_norm": 0.6262525916099548, | |
| "learning_rate": 2.7419354838709676e-06, | |
| "loss": 1.0282, | |
| "num_tokens": 25286383.0, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 1.7741935483870968, | |
| "grad_norm": 0.5740677118301392, | |
| "learning_rate": 2.338709677419355e-06, | |
| "loss": 0.9635, | |
| "num_tokens": 25917151.0, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.814516129032258, | |
| "grad_norm": 0.5970714688301086, | |
| "learning_rate": 1.935483870967742e-06, | |
| "loss": 1.0004, | |
| "num_tokens": 26525780.0, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 1.8548387096774195, | |
| "grad_norm": 0.6618486642837524, | |
| "learning_rate": 1.5322580645161292e-06, | |
| "loss": 0.9982, | |
| "num_tokens": 27096606.0, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.8951612903225805, | |
| "grad_norm": 0.5806471705436707, | |
| "learning_rate": 1.1290322580645162e-06, | |
| "loss": 1.0286, | |
| "num_tokens": 27687166.0, | |
| "step": 235 | |
| }, | |
| { | |
| "epoch": 1.935483870967742, | |
| "grad_norm": 0.6013484001159668, | |
| "learning_rate": 7.258064516129033e-07, | |
| "loss": 0.9882, | |
| "num_tokens": 28284352.0, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.9758064516129032, | |
| "grad_norm": 0.5959106683731079, | |
| "learning_rate": 3.2258064516129035e-07, | |
| "loss": 0.9685, | |
| "num_tokens": 28878645.0, | |
| "step": 245 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "num_tokens": 29227434.0, | |
| "step": 248, | |
| "total_flos": 4.8695749825160806e+17, | |
| "train_loss": 1.088731300446295, | |
| "train_runtime": 315.8129, | |
| "train_samples_per_second": 100.097, | |
| "train_steps_per_second": 0.785 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 248, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 4.8695749825160806e+17, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |