| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.902439024390244, | |
| "eval_steps": 500, | |
| "global_step": 200, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 5.21875, | |
| "learning_rate": 2.5e-06, | |
| "loss": 6.2944, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 5.75, | |
| "learning_rate": 2.9968542393565676e-06, | |
| "loss": 8.1888, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 4.59375, | |
| "learning_rate": 2.984097196854534e-06, | |
| "loss": 4.2333, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "grad_norm": 4.09375, | |
| "learning_rate": 2.9616157869703894e-06, | |
| "loss": 3.6424, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "grad_norm": 4.1875, | |
| "learning_rate": 2.929557316423291e-06, | |
| "loss": 5.3747, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 3.453125, | |
| "learning_rate": 2.8881318444640566e-06, | |
| "loss": 6.7731, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "grad_norm": 3.984375, | |
| "learning_rate": 2.8376108064873215e-06, | |
| "loss": 6.4852, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 4.1875, | |
| "learning_rate": 2.778325235483954e-06, | |
| "loss": 6.5685, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 3.078125, | |
| "learning_rate": 2.710663592987414e-06, | |
| "loss": 7.1279, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "grad_norm": 3.21875, | |
| "learning_rate": 2.6350692237265428e-06, | |
| "loss": 3.2127, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "grad_norm": 2.859375, | |
| "learning_rate": 2.552037450662849e-06, | |
| "loss": 9.9494, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "grad_norm": 3.078125, | |
| "learning_rate": 2.4621123294467098e-06, | |
| "loss": 5.1331, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "grad_norm": 2.96875, | |
| "learning_rate": 2.3658830835585295e-06, | |
| "loss": 5.8588, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 1.37, | |
| "grad_norm": 2.953125, | |
| "learning_rate": 2.2639802434931445e-06, | |
| "loss": 4.6236, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.46, | |
| "grad_norm": 3.09375, | |
| "learning_rate": 2.1570715152850234e-06, | |
| "loss": 3.4357, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.56, | |
| "grad_norm": 2.953125, | |
| "learning_rate": 2.0458574054452316e-06, | |
| "loss": 5.1455, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "grad_norm": 3.1875, | |
| "learning_rate": 1.9310666309772426e-06, | |
| "loss": 2.8809, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.76, | |
| "grad_norm": 3.0625, | |
| "learning_rate": 1.813451344546913e-06, | |
| "loss": 8.66, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.85, | |
| "grad_norm": 3.109375, | |
| "learning_rate": 1.6937822060931097e-06, | |
| "loss": 6.1746, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.95, | |
| "grad_norm": 3.359375, | |
| "learning_rate": 1.5728433331716726e-06, | |
| "loss": 7.1489, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "grad_norm": 3.453125, | |
| "learning_rate": 1.4514271631199665e-06, | |
| "loss": 9.0421, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 2.15, | |
| "grad_norm": 3.328125, | |
| "learning_rate": 1.3303292607070737e-06, | |
| "loss": 3.6143, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 2.24, | |
| "grad_norm": 3.5625, | |
| "learning_rate": 1.2103431052918681e-06, | |
| "loss": 5.9441, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 2.34, | |
| "grad_norm": 2.875, | |
| "learning_rate": 1.0922548916454855e-06, | |
| "loss": 3.1449, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 2.44, | |
| "grad_norm": 3.078125, | |
| "learning_rate": 9.768383785051766e-07, | |
| "loss": 8.4543, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 2.54, | |
| "grad_norm": 3.734375, | |
| "learning_rate": 8.648498186137653e-07, | |
| "loss": 6.5607, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 2.63, | |
| "grad_norm": 3.328125, | |
| "learning_rate": 7.570230034650192e-07, | |
| "loss": 8.5402, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 2.73, | |
| "grad_norm": 3.09375, | |
| "learning_rate": 6.540644552236401e-07, | |
| "loss": 2.9279, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 2.83, | |
| "grad_norm": 3.171875, | |
| "learning_rate": 5.566487973242332e-07, | |
| "loss": 6.1072, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 2.93, | |
| "grad_norm": 2.53125, | |
| "learning_rate": 4.6541433408284356e-07, | |
| "loss": 5.8211, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "grad_norm": 2.765625, | |
| "learning_rate": 3.8095886828513094e-07, | |
| "loss": 9.2977, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "grad_norm": 3.34375, | |
| "learning_rate": 3.0383578415591913e-07, | |
| "loss": 3.2964, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 3.22, | |
| "grad_norm": 3.53125, | |
| "learning_rate": 2.345504213759934e-07, | |
| "loss": 5.6876, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 3.32, | |
| "grad_norm": 2.78125, | |
| "learning_rate": 1.7355676390496482e-07, | |
| "loss": 3.7788, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 3.41, | |
| "grad_norm": 3.265625, | |
| "learning_rate": 1.2125446530628187e-07, | |
| "loss": 3.0528, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 3.51, | |
| "grad_norm": 3.859375, | |
| "learning_rate": 7.798623006559436e-08, | |
| "loss": 27.2733, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 3.61, | |
| "grad_norm": 3.03125, | |
| "learning_rate": 4.4035568061061235e-08, | |
| "loss": 6.5668, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 3.71, | |
| "grad_norm": 2.609375, | |
| "learning_rate": 1.962493689916395e-08, | |
| "loss": 3.7337, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 3.8, | |
| "grad_norm": 2.796875, | |
| "learning_rate": 4.9142842881459425e-09, | |
| "loss": 3.2751, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 3.9, | |
| "grad_norm": 3.3125, | |
| "learning_rate": 0.0, | |
| "loss": 8.9476, | |
| "step": 200 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 200, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 4, | |
| "save_steps": 20, | |
| "total_flos": 7.993988039172096e+16, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |