| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.8021265092809515, | |
| "eval_steps": 500, | |
| "global_step": 20000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.04505316273202379, | |
| "grad_norm": 0.0, | |
| "learning_rate": 1.25e-05, | |
| "loss": 1347.2, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.09010632546404758, | |
| "grad_norm": 3153.3505859375, | |
| "learning_rate": 2.5e-05, | |
| "loss": 477.367, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.13515948819607138, | |
| "grad_norm": 1858.1285400390625, | |
| "learning_rate": 3.7500000000000003e-05, | |
| "loss": 442.957, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.18021265092809516, | |
| "grad_norm": 2545.223388671875, | |
| "learning_rate": 5e-05, | |
| "loss": 433.9167, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.22526581366011894, | |
| "grad_norm": 2296.039794921875, | |
| "learning_rate": 4.990486745229364e-05, | |
| "loss": 418.3493, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.27031897639214275, | |
| "grad_norm": 2642.889404296875, | |
| "learning_rate": 4.962019382530521e-05, | |
| "loss": 407.9998, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.3153721391241665, | |
| "grad_norm": 0.0, | |
| "learning_rate": 4.914814565722671e-05, | |
| "loss": 410.2617, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.3604253018561903, | |
| "grad_norm": 2563.190673828125, | |
| "learning_rate": 4.849231551964771e-05, | |
| "loss": 397.9934, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.40547846458821407, | |
| "grad_norm": 3283.779052734375, | |
| "learning_rate": 4.765769467591625e-05, | |
| "loss": 417.8361, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.4505316273202379, | |
| "grad_norm": 2174.240234375, | |
| "learning_rate": 4.665063509461097e-05, | |
| "loss": 375.8966, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.4955847900522617, | |
| "grad_norm": 2514.682373046875, | |
| "learning_rate": 4.54788011072248e-05, | |
| "loss": 388.2395, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.5406379527842855, | |
| "grad_norm": 3136.03759765625, | |
| "learning_rate": 4.415111107797445e-05, | |
| "loss": 394.8071, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.5856911155163093, | |
| "grad_norm": 1837.290283203125, | |
| "learning_rate": 4.267766952966369e-05, | |
| "loss": 366.3096, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.630744278248333, | |
| "grad_norm": 2366.833740234375, | |
| "learning_rate": 4.1069690242163484e-05, | |
| "loss": 367.1677, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.6757974409803568, | |
| "grad_norm": 2263.440673828125, | |
| "learning_rate": 3.933941090877615e-05, | |
| "loss": 372.6355, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.7208506037123806, | |
| "grad_norm": 2522.50244140625, | |
| "learning_rate": 3.7500000000000003e-05, | |
| "loss": 367.763, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.7659037664444044, | |
| "grad_norm": 2936.105712890625, | |
| "learning_rate": 3.556545654351749e-05, | |
| "loss": 354.5393, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.8109569291764281, | |
| "grad_norm": 0.0, | |
| "learning_rate": 3.355050358314172e-05, | |
| "loss": 362.7008, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.856010091908452, | |
| "grad_norm": 2301.9541015625, | |
| "learning_rate": 3.147047612756302e-05, | |
| "loss": 339.9951, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.9010632546404758, | |
| "grad_norm": 2941.145751953125, | |
| "learning_rate": 2.9341204441673266e-05, | |
| "loss": 357.0448, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.9461164173724995, | |
| "grad_norm": 2575.2666015625, | |
| "learning_rate": 2.717889356869146e-05, | |
| "loss": 356.5437, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 0.9911695801045234, | |
| "grad_norm": 3181.543212890625, | |
| "learning_rate": 2.5e-05, | |
| "loss": 350.985, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_loss": 382.581298828125, | |
| "eval_runtime": 227.6629, | |
| "eval_samples_per_second": 43.332, | |
| "eval_steps_per_second": 5.42, | |
| "step": 11098 | |
| }, | |
| { | |
| "epoch": 1.0362227428365471, | |
| "grad_norm": 2919.064208984375, | |
| "learning_rate": 2.2821106431308544e-05, | |
| "loss": 328.9697, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 1.081275905568571, | |
| "grad_norm": 1792.9368896484375, | |
| "learning_rate": 2.0658795558326743e-05, | |
| "loss": 316.0124, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 1.1263290683005946, | |
| "grad_norm": 3145.044921875, | |
| "learning_rate": 1.852952387243698e-05, | |
| "loss": 318.4672, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 1.1713822310326185, | |
| "grad_norm": 2122.60693359375, | |
| "learning_rate": 1.6449496416858284e-05, | |
| "loss": 310.955, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 1.2164353937646424, | |
| "grad_norm": 0.0, | |
| "learning_rate": 1.443454345648252e-05, | |
| "loss": 308.6115, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 1.261488556496666, | |
| "grad_norm": 2367.4453125, | |
| "learning_rate": 1.2500000000000006e-05, | |
| "loss": 308.7791, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 1.30654171922869, | |
| "grad_norm": 3921.755126953125, | |
| "learning_rate": 1.0660589091223855e-05, | |
| "loss": 314.4502, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 1.3515948819607138, | |
| "grad_norm": 3169.499267578125, | |
| "learning_rate": 8.930309757836517e-06, | |
| "loss": 299.5019, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 1.3966480446927374, | |
| "grad_norm": 1866.3717041015625, | |
| "learning_rate": 7.3223304703363135e-06, | |
| "loss": 303.1934, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 1.4417012074247613, | |
| "grad_norm": 2293.599365234375, | |
| "learning_rate": 5.848888922025553e-06, | |
| "loss": 299.5156, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 1.4867543701567851, | |
| "grad_norm": 2268.71142578125, | |
| "learning_rate": 4.521198892775203e-06, | |
| "loss": 307.7869, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 1.5318075328888088, | |
| "grad_norm": 3546.200439453125, | |
| "learning_rate": 3.3493649053890326e-06, | |
| "loss": 290.4306, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 1.5768606956208324, | |
| "grad_norm": 2096.766845703125, | |
| "learning_rate": 2.3423053240837515e-06, | |
| "loss": 299.7911, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 1.6219138583528565, | |
| "grad_norm": 0.0, | |
| "learning_rate": 1.5076844803522922e-06, | |
| "loss": 302.4306, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 1.6669670210848802, | |
| "grad_norm": 1852.839111328125, | |
| "learning_rate": 8.51854342773295e-07, | |
| "loss": 305.6073, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 1.7120201838169038, | |
| "grad_norm": 2610.0205078125, | |
| "learning_rate": 3.7980617469479953e-07, | |
| "loss": 308.6735, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 1.7570733465489279, | |
| "grad_norm": 1428.6187744140625, | |
| "learning_rate": 9.513254770636137e-08, | |
| "loss": 296.7373, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 1.8021265092809515, | |
| "grad_norm": 0.0, | |
| "learning_rate": 0.0, | |
| "loss": 290.4464, | |
| "step": 20000 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 20000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 10000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 0.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |