{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.08333333333333333, "eval_steps": 500, "global_step": 5000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0008333333333333334, "grad_norm": 9.0625, "learning_rate": 0.00019800000000000002, "loss": 0.4344, "step": 50 }, { "epoch": 0.0016666666666666668, "grad_norm": 5.53125, "learning_rate": 0.000196, "loss": 0.1842, "step": 100 }, { "epoch": 0.0025, "grad_norm": 8.875, "learning_rate": 0.000194, "loss": 0.1964, "step": 150 }, { "epoch": 0.0033333333333333335, "grad_norm": 0.0, "learning_rate": 0.000192, "loss": 0.1988, "step": 200 }, { "epoch": 0.004166666666666667, "grad_norm": 2.3125, "learning_rate": 0.00019, "loss": 0.3236, "step": 250 }, { "epoch": 0.005, "grad_norm": 14.5625, "learning_rate": 0.000188, "loss": 0.1435, "step": 300 }, { "epoch": 0.005833333333333334, "grad_norm": 20.125, "learning_rate": 0.00018600000000000002, "loss": 0.173, "step": 350 }, { "epoch": 0.006666666666666667, "grad_norm": 5.375, "learning_rate": 0.00018400000000000003, "loss": 0.2097, "step": 400 }, { "epoch": 0.0075, "grad_norm": 27.0, "learning_rate": 0.000182, "loss": 0.1865, "step": 450 }, { "epoch": 0.008333333333333333, "grad_norm": 0.0, "learning_rate": 0.00018, "loss": 0.1633, "step": 500 }, { "epoch": 0.009166666666666667, "grad_norm": 0.1748046875, "learning_rate": 0.00017800000000000002, "loss": 0.2406, "step": 550 }, { "epoch": 0.01, "grad_norm": 1.4921875, "learning_rate": 0.00017600000000000002, "loss": 0.1364, "step": 600 }, { "epoch": 0.010833333333333334, "grad_norm": 1.9609375, "learning_rate": 0.000174, "loss": 0.2088, "step": 650 }, { "epoch": 0.011666666666666667, "grad_norm": 5.5, "learning_rate": 0.000172, "loss": 0.2133, "step": 700 }, { "epoch": 0.0125, "grad_norm": 6.8125, "learning_rate": 0.00017, "loss": 0.0794, "step": 750 }, { "epoch": 0.013333333333333334, "grad_norm": 1.6640625, "learning_rate": 0.000168, "loss": 0.1982, "step": 800 }, { "epoch": 0.014166666666666666, "grad_norm": 0.0, "learning_rate": 0.000166, "loss": 0.0913, "step": 850 }, { "epoch": 0.015, "grad_norm": 1.5859375, "learning_rate": 0.000164, "loss": 0.1682, "step": 900 }, { "epoch": 0.015833333333333335, "grad_norm": 0.0, "learning_rate": 0.000162, "loss": 0.0817, "step": 950 }, { "epoch": 0.016666666666666666, "grad_norm": 0.046142578125, "learning_rate": 0.00016, "loss": 0.2242, "step": 1000 }, { "epoch": 0.0175, "grad_norm": 6.3125, "learning_rate": 0.00015800000000000002, "loss": 0.2627, "step": 1050 }, { "epoch": 0.018333333333333333, "grad_norm": 15.0, "learning_rate": 0.00015600000000000002, "loss": 0.177, "step": 1100 }, { "epoch": 0.019166666666666665, "grad_norm": 0.45703125, "learning_rate": 0.000154, "loss": 0.0872, "step": 1150 }, { "epoch": 0.02, "grad_norm": 0.0, "learning_rate": 0.000152, "loss": 0.1576, "step": 1200 }, { "epoch": 0.020833333333333332, "grad_norm": 4.9375, "learning_rate": 0.00015000000000000001, "loss": 0.1687, "step": 1250 }, { "epoch": 0.021666666666666667, "grad_norm": 9.4375, "learning_rate": 0.000148, "loss": 0.1967, "step": 1300 }, { "epoch": 0.0225, "grad_norm": 0.00150299072265625, "learning_rate": 0.000146, "loss": 0.1067, "step": 1350 }, { "epoch": 0.023333333333333334, "grad_norm": 2.046875, "learning_rate": 0.000144, "loss": 0.2409, "step": 1400 }, { "epoch": 0.024166666666666666, "grad_norm": 0.0, "learning_rate": 0.000142, "loss": 0.1165, "step": 1450 }, { "epoch": 0.025, "grad_norm": 0.0, "learning_rate": 0.00014, "loss": 0.074, "step": 1500 }, { "epoch": 0.025833333333333333, "grad_norm": 0.00665283203125, "learning_rate": 0.000138, "loss": 0.1285, "step": 1550 }, { "epoch": 0.02666666666666667, "grad_norm": 0.10107421875, "learning_rate": 0.00013600000000000003, "loss": 0.0569, "step": 1600 }, { "epoch": 0.0275, "grad_norm": 0.07421875, "learning_rate": 0.000134, "loss": 0.0937, "step": 1650 }, { "epoch": 0.028333333333333332, "grad_norm": 0.267578125, "learning_rate": 0.000132, "loss": 0.1249, "step": 1700 }, { "epoch": 0.029166666666666667, "grad_norm": 39.75, "learning_rate": 0.00013000000000000002, "loss": 0.1099, "step": 1750 }, { "epoch": 0.03, "grad_norm": 0.328125, "learning_rate": 0.00012800000000000002, "loss": 0.1529, "step": 1800 }, { "epoch": 0.030833333333333334, "grad_norm": 0.328125, "learning_rate": 0.000126, "loss": 0.1359, "step": 1850 }, { "epoch": 0.03166666666666667, "grad_norm": 0.0011138916015625, "learning_rate": 0.000124, "loss": 0.1421, "step": 1900 }, { "epoch": 0.0325, "grad_norm": 13.4375, "learning_rate": 0.000122, "loss": 0.155, "step": 1950 }, { "epoch": 0.03333333333333333, "grad_norm": 2.0625, "learning_rate": 0.00012, "loss": 0.1322, "step": 2000 }, { "epoch": 0.034166666666666665, "grad_norm": 2.5625, "learning_rate": 0.000118, "loss": 0.1956, "step": 2050 }, { "epoch": 0.035, "grad_norm": 0.140625, "learning_rate": 0.000116, "loss": 0.0676, "step": 2100 }, { "epoch": 0.035833333333333335, "grad_norm": 0.0, "learning_rate": 0.00011399999999999999, "loss": 0.1126, "step": 2150 }, { "epoch": 0.03666666666666667, "grad_norm": 0.00421142578125, "learning_rate": 0.00011200000000000001, "loss": 0.0985, "step": 2200 }, { "epoch": 0.0375, "grad_norm": 2.3125, "learning_rate": 0.00011000000000000002, "loss": 0.1099, "step": 2250 }, { "epoch": 0.03833333333333333, "grad_norm": 0.47265625, "learning_rate": 0.00010800000000000001, "loss": 0.2447, "step": 2300 }, { "epoch": 0.03916666666666667, "grad_norm": 0.0, "learning_rate": 0.00010600000000000002, "loss": 0.0887, "step": 2350 }, { "epoch": 0.04, "grad_norm": 9.8125, "learning_rate": 0.00010400000000000001, "loss": 0.1098, "step": 2400 }, { "epoch": 0.04083333333333333, "grad_norm": 0.06201171875, "learning_rate": 0.00010200000000000001, "loss": 0.2096, "step": 2450 }, { "epoch": 0.041666666666666664, "grad_norm": 0.0240478515625, "learning_rate": 0.0001, "loss": 0.0477, "step": 2500 }, { "epoch": 0.0425, "grad_norm": 0.0, "learning_rate": 9.8e-05, "loss": 0.0702, "step": 2550 }, { "epoch": 0.043333333333333335, "grad_norm": 0.08447265625, "learning_rate": 9.6e-05, "loss": 0.122, "step": 2600 }, { "epoch": 0.04416666666666667, "grad_norm": 1.59375, "learning_rate": 9.4e-05, "loss": 0.1273, "step": 2650 }, { "epoch": 0.045, "grad_norm": 0.0, "learning_rate": 9.200000000000001e-05, "loss": 0.0616, "step": 2700 }, { "epoch": 0.04583333333333333, "grad_norm": 0.1259765625, "learning_rate": 9e-05, "loss": 0.0633, "step": 2750 }, { "epoch": 0.04666666666666667, "grad_norm": 0.0, "learning_rate": 8.800000000000001e-05, "loss": 0.0587, "step": 2800 }, { "epoch": 0.0475, "grad_norm": 0.0, "learning_rate": 8.6e-05, "loss": 0.0773, "step": 2850 }, { "epoch": 0.04833333333333333, "grad_norm": 0.27734375, "learning_rate": 8.4e-05, "loss": 0.2962, "step": 2900 }, { "epoch": 0.049166666666666664, "grad_norm": 0.059326171875, "learning_rate": 8.2e-05, "loss": 0.0258, "step": 2950 }, { "epoch": 0.05, "grad_norm": 0.033447265625, "learning_rate": 8e-05, "loss": 0.2214, "step": 3000 }, { "epoch": 0.050833333333333335, "grad_norm": 0.0091552734375, "learning_rate": 7.800000000000001e-05, "loss": 0.0579, "step": 3050 }, { "epoch": 0.051666666666666666, "grad_norm": 0.0, "learning_rate": 7.6e-05, "loss": 0.0433, "step": 3100 }, { "epoch": 0.0525, "grad_norm": 0.03173828125, "learning_rate": 7.4e-05, "loss": 0.1355, "step": 3150 }, { "epoch": 0.05333333333333334, "grad_norm": 20.25, "learning_rate": 7.2e-05, "loss": 0.0966, "step": 3200 }, { "epoch": 0.05416666666666667, "grad_norm": 0.0, "learning_rate": 7e-05, "loss": 0.0395, "step": 3250 }, { "epoch": 0.055, "grad_norm": 10.25, "learning_rate": 6.800000000000001e-05, "loss": 0.1048, "step": 3300 }, { "epoch": 0.05583333333333333, "grad_norm": 0.00151824951171875, "learning_rate": 6.6e-05, "loss": 0.0489, "step": 3350 }, { "epoch": 0.056666666666666664, "grad_norm": 5.71875, "learning_rate": 6.400000000000001e-05, "loss": 0.1054, "step": 3400 }, { "epoch": 0.0575, "grad_norm": 0.004180908203125, "learning_rate": 6.2e-05, "loss": 0.0914, "step": 3450 }, { "epoch": 0.058333333333333334, "grad_norm": 0.7109375, "learning_rate": 6e-05, "loss": 0.1068, "step": 3500 }, { "epoch": 0.059166666666666666, "grad_norm": 0.0, "learning_rate": 5.8e-05, "loss": 0.1098, "step": 3550 }, { "epoch": 0.06, "grad_norm": 0.65234375, "learning_rate": 5.6000000000000006e-05, "loss": 0.128, "step": 3600 }, { "epoch": 0.060833333333333336, "grad_norm": 2.9375, "learning_rate": 5.4000000000000005e-05, "loss": 0.1229, "step": 3650 }, { "epoch": 0.06166666666666667, "grad_norm": 0.01336669921875, "learning_rate": 5.2000000000000004e-05, "loss": 0.2113, "step": 3700 }, { "epoch": 0.0625, "grad_norm": 0.0177001953125, "learning_rate": 5e-05, "loss": 0.1729, "step": 3750 }, { "epoch": 0.06333333333333334, "grad_norm": 0.005767822265625, "learning_rate": 4.8e-05, "loss": 0.0663, "step": 3800 }, { "epoch": 0.06416666666666666, "grad_norm": 1.5, "learning_rate": 4.600000000000001e-05, "loss": 0.0715, "step": 3850 }, { "epoch": 0.065, "grad_norm": 0.130859375, "learning_rate": 4.4000000000000006e-05, "loss": 0.1229, "step": 3900 }, { "epoch": 0.06583333333333333, "grad_norm": 0.9765625, "learning_rate": 4.2e-05, "loss": 0.1127, "step": 3950 }, { "epoch": 0.06666666666666667, "grad_norm": 0.052978515625, "learning_rate": 4e-05, "loss": 0.1435, "step": 4000 }, { "epoch": 0.0675, "grad_norm": 0.0, "learning_rate": 3.8e-05, "loss": 0.0738, "step": 4050 }, { "epoch": 0.06833333333333333, "grad_norm": 5.875, "learning_rate": 3.6e-05, "loss": 0.0345, "step": 4100 }, { "epoch": 0.06916666666666667, "grad_norm": 0.2314453125, "learning_rate": 3.4000000000000007e-05, "loss": 0.1171, "step": 4150 }, { "epoch": 0.07, "grad_norm": 0.0, "learning_rate": 3.2000000000000005e-05, "loss": 0.0412, "step": 4200 }, { "epoch": 0.07083333333333333, "grad_norm": 0.0247802734375, "learning_rate": 3e-05, "loss": 0.0479, "step": 4250 }, { "epoch": 0.07166666666666667, "grad_norm": 0.0, "learning_rate": 2.8000000000000003e-05, "loss": 0.12, "step": 4300 }, { "epoch": 0.0725, "grad_norm": 0.07666015625, "learning_rate": 2.6000000000000002e-05, "loss": 0.1109, "step": 4350 }, { "epoch": 0.07333333333333333, "grad_norm": 14.0625, "learning_rate": 2.4e-05, "loss": 0.0773, "step": 4400 }, { "epoch": 0.07416666666666667, "grad_norm": 6.78125, "learning_rate": 2.2000000000000003e-05, "loss": 0.0825, "step": 4450 }, { "epoch": 0.075, "grad_norm": 1.8046875, "learning_rate": 2e-05, "loss": 0.1683, "step": 4500 }, { "epoch": 0.07583333333333334, "grad_norm": 17.125, "learning_rate": 1.8e-05, "loss": 0.2124, "step": 4550 }, { "epoch": 0.07666666666666666, "grad_norm": 0.10302734375, "learning_rate": 1.6000000000000003e-05, "loss": 0.1058, "step": 4600 }, { "epoch": 0.0775, "grad_norm": 0.07470703125, "learning_rate": 1.4000000000000001e-05, "loss": 0.1423, "step": 4650 }, { "epoch": 0.07833333333333334, "grad_norm": 0.02685546875, "learning_rate": 1.2e-05, "loss": 0.1012, "step": 4700 }, { "epoch": 0.07916666666666666, "grad_norm": 0.1484375, "learning_rate": 1e-05, "loss": 0.05, "step": 4750 }, { "epoch": 0.08, "grad_norm": 8.75, "learning_rate": 8.000000000000001e-06, "loss": 0.0657, "step": 4800 }, { "epoch": 0.08083333333333333, "grad_norm": 0.00335693359375, "learning_rate": 6e-06, "loss": 0.1015, "step": 4850 }, { "epoch": 0.08166666666666667, "grad_norm": 2.84375, "learning_rate": 4.000000000000001e-06, "loss": 0.1311, "step": 4900 }, { "epoch": 0.0825, "grad_norm": 0.953125, "learning_rate": 2.0000000000000003e-06, "loss": 0.1124, "step": 4950 }, { "epoch": 0.08333333333333333, "grad_norm": 0.002471923828125, "learning_rate": 0.0, "loss": 0.1312, "step": 5000 } ], "logging_steps": 50, "max_steps": 5000, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 1000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 9.001157227118592e+16, "train_batch_size": 1, "trial_name": null, "trial_params": null }