| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.974025974025974, | |
| "eval_steps": 500, | |
| "global_step": 38, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05194805194805195, | |
| "grad_norm": 4.959436893463135, | |
| "learning_rate": 2e-05, | |
| "loss": 2.0897, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.1038961038961039, | |
| "grad_norm": 3.942784547805786, | |
| "learning_rate": 4e-05, | |
| "loss": 2.3866, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.15584415584415584, | |
| "grad_norm": 4.221513748168945, | |
| "learning_rate": 6e-05, | |
| "loss": 2.4587, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.2077922077922078, | |
| "grad_norm": 3.284791946411133, | |
| "learning_rate": 8e-05, | |
| "loss": 2.1799, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.2597402597402597, | |
| "grad_norm": 5.71047306060791, | |
| "learning_rate": 0.0001, | |
| "loss": 2.1779, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.3116883116883117, | |
| "grad_norm": 3.1526741981506348, | |
| "learning_rate": 0.00012, | |
| "loss": 2.1932, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.36363636363636365, | |
| "grad_norm": 3.1342685222625732, | |
| "learning_rate": 0.00014, | |
| "loss": 1.4442, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.4155844155844156, | |
| "grad_norm": 5.338088035583496, | |
| "learning_rate": 0.00016, | |
| "loss": 2.0, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.4675324675324675, | |
| "grad_norm": 4.116346836090088, | |
| "learning_rate": 0.00018, | |
| "loss": 1.5286, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.5194805194805194, | |
| "grad_norm": 2.5058088302612305, | |
| "learning_rate": 0.0002, | |
| "loss": 1.4238, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.5714285714285714, | |
| "grad_norm": 4.636209487915039, | |
| "learning_rate": 0.0001999317060143023, | |
| "loss": 1.511, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.6233766233766234, | |
| "grad_norm": 3.1061031818389893, | |
| "learning_rate": 0.00019972691733857883, | |
| "loss": 1.3241, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.6753246753246753, | |
| "grad_norm": 2.3636441230773926, | |
| "learning_rate": 0.0001993859136895274, | |
| "loss": 1.044, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.7272727272727273, | |
| "grad_norm": 2.5445024967193604, | |
| "learning_rate": 0.0001989091608371146, | |
| "loss": 0.9548, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.7792207792207793, | |
| "grad_norm": 2.371872901916504, | |
| "learning_rate": 0.0001982973099683902, | |
| "loss": 1.0614, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.8311688311688312, | |
| "grad_norm": 5.0257039070129395, | |
| "learning_rate": 0.00019755119679804367, | |
| "loss": 1.1408, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.8831168831168831, | |
| "grad_norm": 1.9052318334579468, | |
| "learning_rate": 0.00019667184042691875, | |
| "loss": 0.8993, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.935064935064935, | |
| "grad_norm": 2.2592344284057617, | |
| "learning_rate": 0.0001956604419500441, | |
| "loss": 1.0319, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.987012987012987, | |
| "grad_norm": 2.264328718185425, | |
| "learning_rate": 0.00019451838281608197, | |
| "loss": 1.483, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 1.0389610389610389, | |
| "grad_norm": 4.711090564727783, | |
| "learning_rate": 0.00019324722294043558, | |
| "loss": 1.4984, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 1.0909090909090908, | |
| "grad_norm": 1.5374799966812134, | |
| "learning_rate": 0.00019184869857459232, | |
| "loss": 0.7076, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 1.1428571428571428, | |
| "grad_norm": 2.26493501663208, | |
| "learning_rate": 0.0001903247199346129, | |
| "loss": 1.1543, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 1.1948051948051948, | |
| "grad_norm": 1.51936674118042, | |
| "learning_rate": 0.0001886773685920062, | |
| "loss": 0.7064, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 1.2467532467532467, | |
| "grad_norm": 2.1054816246032715, | |
| "learning_rate": 0.00018690889463055283, | |
| "loss": 0.8401, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 1.2987012987012987, | |
| "grad_norm": 2.4671549797058105, | |
| "learning_rate": 0.00018502171357296144, | |
| "loss": 1.0385, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 1.3506493506493507, | |
| "grad_norm": 1.6618045568466187, | |
| "learning_rate": 0.00018301840308155507, | |
| "loss": 0.5543, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 1.4025974025974026, | |
| "grad_norm": 1.7877517938613892, | |
| "learning_rate": 0.00018090169943749476, | |
| "loss": 0.5464, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 1.4545454545454546, | |
| "grad_norm": 3.1368205547332764, | |
| "learning_rate": 0.00017867449380334834, | |
| "loss": 1.2296, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 1.5064935064935066, | |
| "grad_norm": 1.902140498161316, | |
| "learning_rate": 0.00017633982827411032, | |
| "loss": 0.8562, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 1.5584415584415585, | |
| "grad_norm": 1.6555312871932983, | |
| "learning_rate": 0.00017390089172206592, | |
| "loss": 0.7186, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 1.6103896103896105, | |
| "grad_norm": 1.526365041732788, | |
| "learning_rate": 0.00017136101544117525, | |
| "loss": 0.6501, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 1.6623376623376624, | |
| "grad_norm": 1.7592672109603882, | |
| "learning_rate": 0.00016872366859692627, | |
| "loss": 0.9721, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 1.7142857142857144, | |
| "grad_norm": 1.7448221445083618, | |
| "learning_rate": 0.0001659924534878723, | |
| "loss": 0.6916, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 1.7662337662337664, | |
| "grad_norm": 2.216275930404663, | |
| "learning_rate": 0.0001631711006253251, | |
| "loss": 0.8664, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 1.8181818181818183, | |
| "grad_norm": 1.5474787950515747, | |
| "learning_rate": 0.00016026346363792567, | |
| "loss": 0.5205, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 1.87012987012987, | |
| "grad_norm": 2.3516626358032227, | |
| "learning_rate": 0.00015727351400805052, | |
| "loss": 1.0247, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 1.922077922077922, | |
| "grad_norm": 1.921420931816101, | |
| "learning_rate": 0.00015420533564724495, | |
| "loss": 0.9075, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 1.974025974025974, | |
| "grad_norm": 1.9142687320709229, | |
| "learning_rate": 0.0001510631193180907, | |
| "loss": 0.8456, | |
| "step": 38 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 95, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 19, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.261543204281385e+17, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |