{ "best_metric": 0.2015175074338913, "best_model_checkpoint": "./output_solor/exp_16/checkpoint-95", "epoch": 5.0, "eval_steps": 5, "global_step": 95, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.05, "learning_rate": 5.0000000000000004e-08, "loss": 1.3493, "step": 1 }, { "epoch": 0.05, "eval_loss": 1.2794537544250488, "eval_runtime": 20.3935, "eval_samples_per_second": 3.138, "eval_steps_per_second": 0.392, "step": 1 }, { "epoch": 0.11, "learning_rate": 2.0000000000000002e-07, "loss": 1.301, "step": 2 }, { "epoch": 0.16, "learning_rate": 4.5e-07, "loss": 1.2596, "step": 3 }, { "epoch": 0.21, "learning_rate": 8.000000000000001e-07, "loss": 1.2522, "step": 4 }, { "epoch": 0.26, "learning_rate": 1.2500000000000003e-06, "loss": 1.2483, "step": 5 }, { "epoch": 0.26, "eval_loss": 1.2768707275390625, "eval_runtime": 20.6266, "eval_samples_per_second": 3.103, "eval_steps_per_second": 0.388, "step": 5 }, { "epoch": 0.32, "learning_rate": 1.8e-06, "loss": 1.3056, "step": 6 }, { "epoch": 0.37, "learning_rate": 2.4500000000000003e-06, "loss": 1.4414, "step": 7 }, { "epoch": 0.42, "learning_rate": 3.2000000000000003e-06, "loss": 1.2347, "step": 8 }, { "epoch": 0.47, "learning_rate": 4.05e-06, "loss": 1.2399, "step": 9 }, { "epoch": 0.53, "learning_rate": 5.000000000000001e-06, "loss": 1.2275, "step": 10 }, { "epoch": 0.53, "eval_loss": 1.2099063396453857, "eval_runtime": 20.6171, "eval_samples_per_second": 3.104, "eval_steps_per_second": 0.388, "step": 10 }, { "epoch": 0.58, "learning_rate": 6.05e-06, "loss": 1.154, "step": 11 }, { "epoch": 0.63, "learning_rate": 7.2e-06, "loss": 1.2325, "step": 12 }, { "epoch": 0.68, "learning_rate": 8.45e-06, "loss": 1.1022, "step": 13 }, { "epoch": 0.74, "learning_rate": 9.800000000000001e-06, "loss": 1.0295, "step": 14 }, { "epoch": 0.79, "learning_rate": 1.1249999999999999e-05, "loss": 1.0529, "step": 15 }, { "epoch": 0.79, "eval_loss": 1.0723849534988403, "eval_runtime": 20.6213, "eval_samples_per_second": 3.104, "eval_steps_per_second": 0.388, "step": 15 }, { "epoch": 0.84, "learning_rate": 1.2800000000000001e-05, "loss": 1.1019, "step": 16 }, { "epoch": 0.89, "learning_rate": 1.4450000000000002e-05, "loss": 1.0158, "step": 17 }, { "epoch": 0.95, "learning_rate": 1.62e-05, "loss": 1.0235, "step": 18 }, { "epoch": 1.0, "learning_rate": 1.805e-05, "loss": 1.1272, "step": 19 }, { "epoch": 1.05, "learning_rate": 2.0000000000000005e-05, "loss": 0.8642, "step": 20 }, { "epoch": 1.05, "eval_loss": 0.9709420204162598, "eval_runtime": 20.3773, "eval_samples_per_second": 3.141, "eval_steps_per_second": 0.393, "step": 20 }, { "epoch": 1.11, "learning_rate": 2.2049999999999997e-05, "loss": 0.996, "step": 21 }, { "epoch": 1.16, "learning_rate": 2.42e-05, "loss": 0.8983, "step": 22 }, { "epoch": 1.21, "learning_rate": 2.6450000000000003e-05, "loss": 0.9876, "step": 23 }, { "epoch": 1.26, "learning_rate": 2.88e-05, "loss": 0.7844, "step": 24 }, { "epoch": 1.32, "learning_rate": 3.125e-05, "loss": 0.8477, "step": 25 }, { "epoch": 1.32, "eval_loss": 0.8244509696960449, "eval_runtime": 20.6023, "eval_samples_per_second": 3.106, "eval_steps_per_second": 0.388, "step": 25 }, { "epoch": 1.37, "learning_rate": 3.38e-05, "loss": 0.786, "step": 26 }, { "epoch": 1.42, "learning_rate": 3.6450000000000005e-05, "loss": 0.6724, "step": 27 }, { "epoch": 1.47, "learning_rate": 3.9200000000000004e-05, "loss": 0.8566, "step": 28 }, { "epoch": 1.53, "learning_rate": 4.205e-05, "loss": 0.6692, "step": 29 }, { "epoch": 1.58, "learning_rate": 4.4999999999999996e-05, "loss": 0.7207, "step": 30 }, { "epoch": 1.58, "eval_loss": 0.6993950009346008, "eval_runtime": 20.6231, "eval_samples_per_second": 3.103, "eval_steps_per_second": 0.388, "step": 30 }, { "epoch": 1.63, "learning_rate": 4.805e-05, "loss": 0.714, "step": 31 }, { "epoch": 1.68, "learning_rate": 5.1200000000000004e-05, "loss": 0.6141, "step": 32 }, { "epoch": 1.74, "learning_rate": 5.445000000000001e-05, "loss": 0.7061, "step": 33 }, { "epoch": 1.79, "learning_rate": 5.780000000000001e-05, "loss": 0.652, "step": 34 }, { "epoch": 1.84, "learning_rate": 6.125e-05, "loss": 0.4656, "step": 35 }, { "epoch": 1.84, "eval_loss": 0.5878281593322754, "eval_runtime": 20.6201, "eval_samples_per_second": 3.104, "eval_steps_per_second": 0.388, "step": 35 }, { "epoch": 1.89, "learning_rate": 6.48e-05, "loss": 0.6225, "step": 36 }, { "epoch": 1.95, "learning_rate": 6.845e-05, "loss": 0.4638, "step": 37 }, { "epoch": 2.0, "learning_rate": 7.22e-05, "loss": 0.595, "step": 38 }, { "epoch": 2.05, "learning_rate": 7.605e-05, "loss": 0.5618, "step": 39 }, { "epoch": 2.11, "learning_rate": 8.000000000000002e-05, "loss": 0.4949, "step": 40 }, { "epoch": 2.11, "eval_loss": 0.4970114827156067, "eval_runtime": 20.6228, "eval_samples_per_second": 3.103, "eval_steps_per_second": 0.388, "step": 40 }, { "epoch": 2.16, "learning_rate": 8.404999999999998e-05, "loss": 0.5576, "step": 41 }, { "epoch": 2.21, "learning_rate": 8.819999999999999e-05, "loss": 0.5537, "step": 42 }, { "epoch": 2.26, "learning_rate": 9.245e-05, "loss": 0.3734, "step": 43 }, { "epoch": 2.32, "learning_rate": 9.68e-05, "loss": 0.4103, "step": 44 }, { "epoch": 2.37, "learning_rate": 0.00010125000000000001, "loss": 0.3497, "step": 45 }, { "epoch": 2.37, "eval_loss": 0.42206770181655884, "eval_runtime": 20.6133, "eval_samples_per_second": 3.105, "eval_steps_per_second": 0.388, "step": 45 }, { "epoch": 2.42, "learning_rate": 0.00010580000000000001, "loss": 0.4948, "step": 46 }, { "epoch": 2.47, "learning_rate": 0.00011045, "loss": 0.2884, "step": 47 }, { "epoch": 2.53, "learning_rate": 0.0001152, "loss": 0.4234, "step": 48 }, { "epoch": 2.58, "learning_rate": 0.00012004999999999999, "loss": 0.3352, "step": 49 }, { "epoch": 2.63, "learning_rate": 0.000125, "loss": 0.3288, "step": 50 }, { "epoch": 2.63, "eval_loss": 0.3672243356704712, "eval_runtime": 20.6204, "eval_samples_per_second": 3.104, "eval_steps_per_second": 0.388, "step": 50 }, { "epoch": 2.68, "learning_rate": 0.00013005, "loss": 0.3481, "step": 51 }, { "epoch": 2.74, "learning_rate": 0.0001352, "loss": 0.362, "step": 52 }, { "epoch": 2.79, "learning_rate": 0.00014045000000000003, "loss": 0.344, "step": 53 }, { "epoch": 2.84, "learning_rate": 0.00014580000000000002, "loss": 0.426, "step": 54 }, { "epoch": 2.89, "learning_rate": 0.00015125000000000002, "loss": 0.3011, "step": 55 }, { "epoch": 2.89, "eval_loss": 0.32503682374954224, "eval_runtime": 20.6368, "eval_samples_per_second": 3.101, "eval_steps_per_second": 0.388, "step": 55 }, { "epoch": 2.95, "learning_rate": 0.00015680000000000002, "loss": 0.4011, "step": 56 }, { "epoch": 3.0, "learning_rate": 0.00016245, "loss": 0.3796, "step": 57 }, { "epoch": 3.05, "learning_rate": 0.0001682, "loss": 0.2726, "step": 58 }, { "epoch": 3.11, "learning_rate": 0.00017404999999999998, "loss": 0.3218, "step": 59 }, { "epoch": 3.16, "learning_rate": 0.00017999999999999998, "loss": 0.2648, "step": 60 }, { "epoch": 3.16, "eval_loss": 0.29000362753868103, "eval_runtime": 20.595, "eval_samples_per_second": 3.108, "eval_steps_per_second": 0.388, "step": 60 }, { "epoch": 3.21, "learning_rate": 0.00018605, "loss": 0.331, "step": 61 }, { "epoch": 3.26, "learning_rate": 0.0001922, "loss": 0.2442, "step": 62 }, { "epoch": 3.32, "learning_rate": 0.00019845000000000003, "loss": 0.2839, "step": 63 }, { "epoch": 3.37, "learning_rate": 0.00020480000000000002, "loss": 0.3206, "step": 64 }, { "epoch": 3.42, "learning_rate": 0.00021125, "loss": 0.3084, "step": 65 }, { "epoch": 3.42, "eval_loss": 0.25911369919776917, "eval_runtime": 20.6071, "eval_samples_per_second": 3.106, "eval_steps_per_second": 0.388, "step": 65 }, { "epoch": 3.47, "learning_rate": 0.00021780000000000004, "loss": 0.2479, "step": 66 }, { "epoch": 3.53, "learning_rate": 0.00022445000000000003, "loss": 0.3003, "step": 67 }, { "epoch": 3.58, "learning_rate": 0.00023120000000000004, "loss": 0.2757, "step": 68 }, { "epoch": 3.63, "learning_rate": 0.00023804999999999996, "loss": 0.1921, "step": 69 }, { "epoch": 3.68, "learning_rate": 0.000245, "loss": 0.2696, "step": 70 }, { "epoch": 3.68, "eval_loss": 0.24587483704090118, "eval_runtime": 20.6172, "eval_samples_per_second": 3.104, "eval_steps_per_second": 0.388, "step": 70 }, { "epoch": 3.74, "learning_rate": 0.00025205000000000003, "loss": 0.2596, "step": 71 }, { "epoch": 3.79, "learning_rate": 0.0002592, "loss": 0.2386, "step": 72 }, { "epoch": 3.84, "learning_rate": 0.00026644999999999994, "loss": 0.2281, "step": 73 }, { "epoch": 3.89, "learning_rate": 0.0002738, "loss": 0.2538, "step": 74 }, { "epoch": 3.95, "learning_rate": 0.00028125000000000003, "loss": 0.2197, "step": 75 }, { "epoch": 3.95, "eval_loss": 0.22858355939388275, "eval_runtime": 20.6122, "eval_samples_per_second": 3.105, "eval_steps_per_second": 0.388, "step": 75 }, { "epoch": 4.0, "learning_rate": 0.0002888, "loss": 0.182, "step": 76 }, { "epoch": 4.05, "learning_rate": 0.00029645, "loss": 0.1985, "step": 77 }, { "epoch": 4.11, "learning_rate": 0.0003042, "loss": 0.1978, "step": 78 }, { "epoch": 4.16, "learning_rate": 0.0003120500000000001, "loss": 0.2283, "step": 79 }, { "epoch": 4.21, "learning_rate": 0.0003200000000000001, "loss": 0.1905, "step": 80 }, { "epoch": 4.21, "eval_loss": 0.21105319261550903, "eval_runtime": 20.595, "eval_samples_per_second": 3.108, "eval_steps_per_second": 0.388, "step": 80 }, { "epoch": 4.26, "learning_rate": 0.0003280500000000001, "loss": 0.2222, "step": 81 }, { "epoch": 4.32, "learning_rate": 0.00033619999999999993, "loss": 0.2051, "step": 82 }, { "epoch": 4.37, "learning_rate": 0.00034445, "loss": 0.2013, "step": 83 }, { "epoch": 4.42, "learning_rate": 0.00035279999999999996, "loss": 0.1751, "step": 84 }, { "epoch": 4.47, "learning_rate": 0.00036124999999999997, "loss": 0.1815, "step": 85 }, { "epoch": 4.47, "eval_loss": 0.2084398865699768, "eval_runtime": 20.6383, "eval_samples_per_second": 3.101, "eval_steps_per_second": 0.388, "step": 85 }, { "epoch": 4.53, "learning_rate": 0.0003698, "loss": 0.1252, "step": 86 }, { "epoch": 4.58, "learning_rate": 0.00037845, "loss": 0.2354, "step": 87 }, { "epoch": 4.63, "learning_rate": 0.0003872, "loss": 0.1651, "step": 88 }, { "epoch": 4.68, "learning_rate": 0.00039605, "loss": 0.2001, "step": 89 }, { "epoch": 4.74, "learning_rate": 0.00040500000000000003, "loss": 0.2164, "step": 90 }, { "epoch": 4.74, "eval_loss": 0.21278557181358337, "eval_runtime": 20.6294, "eval_samples_per_second": 3.102, "eval_steps_per_second": 0.388, "step": 90 }, { "epoch": 4.79, "learning_rate": 0.00041405000000000006, "loss": 0.155, "step": 91 }, { "epoch": 4.84, "learning_rate": 0.00042320000000000004, "loss": 0.2454, "step": 92 }, { "epoch": 4.89, "learning_rate": 0.0004324500000000001, "loss": 0.1804, "step": 93 }, { "epoch": 4.95, "learning_rate": 0.0004418, "loss": 0.2498, "step": 94 }, { "epoch": 5.0, "learning_rate": 0.00045125, "loss": 0.1412, "step": 95 }, { "epoch": 5.0, "eval_loss": 0.2015175074338913, "eval_runtime": 20.5945, "eval_samples_per_second": 3.108, "eval_steps_per_second": 0.388, "step": 95 } ], "logging_steps": 1, "max_steps": 95, "num_input_tokens_seen": 0, "num_train_epochs": 5, "save_steps": 19, "total_flos": 9751829478899712.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }