| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.1227285179435668, | |
| "eval_steps": 500, | |
| "global_step": 84, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.001461053785042462, | |
| "grad_norm": 0.30625924468040466, | |
| "learning_rate": 2.0000000000000002e-07, | |
| "loss": 2.1669, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.002922107570084924, | |
| "grad_norm": 0.29808640480041504, | |
| "learning_rate": 4.0000000000000003e-07, | |
| "loss": 2.0395, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.004383161355127386, | |
| "grad_norm": 0.24509362876415253, | |
| "learning_rate": 6.000000000000001e-07, | |
| "loss": 1.8973, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.005844215140169848, | |
| "grad_norm": 0.27399635314941406, | |
| "learning_rate": 8.000000000000001e-07, | |
| "loss": 2.0403, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.00730526892521231, | |
| "grad_norm": 0.30136558413505554, | |
| "learning_rate": 1.0000000000000002e-06, | |
| "loss": 2.0947, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.008766322710254772, | |
| "grad_norm": 0.30337047576904297, | |
| "learning_rate": 1.2000000000000002e-06, | |
| "loss": 2.0886, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.010227376495297234, | |
| "grad_norm": 0.28456181287765503, | |
| "learning_rate": 1.4000000000000001e-06, | |
| "loss": 1.988, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.011688430280339696, | |
| "grad_norm": 0.2553723454475403, | |
| "learning_rate": 1.6000000000000001e-06, | |
| "loss": 2.0434, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.013149484065382157, | |
| "grad_norm": 0.2616969645023346, | |
| "learning_rate": 1.8000000000000001e-06, | |
| "loss": 1.9692, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.01461053785042462, | |
| "grad_norm": 0.2795903980731964, | |
| "learning_rate": 2.0000000000000003e-06, | |
| "loss": 1.9994, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.01607159163546708, | |
| "grad_norm": 0.27633437514305115, | |
| "learning_rate": 2.2e-06, | |
| "loss": 1.9672, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.017532645420509543, | |
| "grad_norm": 0.3074289858341217, | |
| "learning_rate": 2.4000000000000003e-06, | |
| "loss": 2.0757, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.018993699205552005, | |
| "grad_norm": 0.27444779872894287, | |
| "learning_rate": 2.6e-06, | |
| "loss": 1.9697, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.020454752990594467, | |
| "grad_norm": 0.28576454520225525, | |
| "learning_rate": 2.8000000000000003e-06, | |
| "loss": 2.0516, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.02191580677563693, | |
| "grad_norm": 0.25402647256851196, | |
| "learning_rate": 3e-06, | |
| "loss": 1.9916, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.02337686056067939, | |
| "grad_norm": 0.27070897817611694, | |
| "learning_rate": 3.2000000000000003e-06, | |
| "loss": 2.0079, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.024837914345721853, | |
| "grad_norm": 0.26685598492622375, | |
| "learning_rate": 3.4000000000000005e-06, | |
| "loss": 1.997, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.026298968130764315, | |
| "grad_norm": 0.2818976044654846, | |
| "learning_rate": 3.6000000000000003e-06, | |
| "loss": 2.0454, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.027760021915806777, | |
| "grad_norm": 0.2832131087779999, | |
| "learning_rate": 3.8000000000000005e-06, | |
| "loss": 2.0227, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.02922107570084924, | |
| "grad_norm": 0.28652554750442505, | |
| "learning_rate": 4.000000000000001e-06, | |
| "loss": 2.0293, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.0306821294858917, | |
| "grad_norm": 0.28600171208381653, | |
| "learning_rate": 4.2000000000000004e-06, | |
| "loss": 2.0214, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.03214318327093416, | |
| "grad_norm": 0.2540723383426666, | |
| "learning_rate": 4.4e-06, | |
| "loss": 1.998, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.03360423705597662, | |
| "grad_norm": 0.2570485472679138, | |
| "learning_rate": 4.600000000000001e-06, | |
| "loss": 1.9908, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.03506529084101909, | |
| "grad_norm": 0.2579878866672516, | |
| "learning_rate": 4.800000000000001e-06, | |
| "loss": 1.9355, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.036526344626061545, | |
| "grad_norm": 0.29550811648368835, | |
| "learning_rate": 5e-06, | |
| "loss": 2.0992, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.03798739841110401, | |
| "grad_norm": 0.2850530445575714, | |
| "learning_rate": 5.2e-06, | |
| "loss": 2.042, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.03944845219614647, | |
| "grad_norm": 0.2779623568058014, | |
| "learning_rate": 5.400000000000001e-06, | |
| "loss": 1.9592, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.040909505981188934, | |
| "grad_norm": 0.2843893766403198, | |
| "learning_rate": 5.600000000000001e-06, | |
| "loss": 2.0435, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.04237055976623139, | |
| "grad_norm": 0.26713600754737854, | |
| "learning_rate": 5.8e-06, | |
| "loss": 2.0014, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.04383161355127386, | |
| "grad_norm": 0.2971515357494354, | |
| "learning_rate": 6e-06, | |
| "loss": 2.0246, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.04529266733631632, | |
| "grad_norm": 0.2626894414424896, | |
| "learning_rate": 6.200000000000001e-06, | |
| "loss": 2.0402, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.04675372112135878, | |
| "grad_norm": 0.27606457471847534, | |
| "learning_rate": 6.4000000000000006e-06, | |
| "loss": 2.0549, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.04821477490640124, | |
| "grad_norm": 0.27802756428718567, | |
| "learning_rate": 6.600000000000001e-06, | |
| "loss": 1.9889, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.049675828691443706, | |
| "grad_norm": 0.2772068977355957, | |
| "learning_rate": 6.800000000000001e-06, | |
| "loss": 1.9913, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.051136882476486165, | |
| "grad_norm": 0.2816425561904907, | |
| "learning_rate": 7e-06, | |
| "loss": 1.9677, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.05259793626152863, | |
| "grad_norm": 0.2521456182003021, | |
| "learning_rate": 7.2000000000000005e-06, | |
| "loss": 1.9705, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.05405899004657109, | |
| "grad_norm": 0.24795399606227875, | |
| "learning_rate": 7.4e-06, | |
| "loss": 2.0483, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.055520043831613554, | |
| "grad_norm": 0.2924613654613495, | |
| "learning_rate": 7.600000000000001e-06, | |
| "loss": 2.0585, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.05698109761665601, | |
| "grad_norm": 0.2479739636182785, | |
| "learning_rate": 7.800000000000002e-06, | |
| "loss": 1.9588, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.05844215140169848, | |
| "grad_norm": 0.27038994431495667, | |
| "learning_rate": 8.000000000000001e-06, | |
| "loss": 1.9146, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.059903205186740936, | |
| "grad_norm": 0.263032466173172, | |
| "learning_rate": 8.2e-06, | |
| "loss": 1.925, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.0613642589717834, | |
| "grad_norm": 0.25826576352119446, | |
| "learning_rate": 8.400000000000001e-06, | |
| "loss": 1.9664, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.06282531275682586, | |
| "grad_norm": 0.3021976351737976, | |
| "learning_rate": 8.6e-06, | |
| "loss": 2.0211, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.06428636654186833, | |
| "grad_norm": 0.2642092704772949, | |
| "learning_rate": 8.8e-06, | |
| "loss": 1.8985, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.06574742032691079, | |
| "grad_norm": 0.2693665623664856, | |
| "learning_rate": 9e-06, | |
| "loss": 2.0221, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.06720847411195324, | |
| "grad_norm": 0.24328897893428802, | |
| "learning_rate": 9.200000000000002e-06, | |
| "loss": 1.9289, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.06866952789699571, | |
| "grad_norm": 0.24804741144180298, | |
| "learning_rate": 9.4e-06, | |
| "loss": 1.9128, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.07013058168203817, | |
| "grad_norm": 0.2629912197589874, | |
| "learning_rate": 9.600000000000001e-06, | |
| "loss": 1.8998, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.07159163546708064, | |
| "grad_norm": 0.22671160101890564, | |
| "learning_rate": 9.800000000000001e-06, | |
| "loss": 1.8634, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.07305268925212309, | |
| "grad_norm": 0.22378858923912048, | |
| "learning_rate": 1e-05, | |
| "loss": 1.9038, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.07451374303716556, | |
| "grad_norm": 0.25769534707069397, | |
| "learning_rate": 1.02e-05, | |
| "loss": 1.8324, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.07597479682220802, | |
| "grad_norm": 0.22694693505764008, | |
| "learning_rate": 1.04e-05, | |
| "loss": 1.8374, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.07743585060725047, | |
| "grad_norm": 0.23865583539009094, | |
| "learning_rate": 1.0600000000000002e-05, | |
| "loss": 1.9129, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.07889690439229294, | |
| "grad_norm": 0.23314256966114044, | |
| "learning_rate": 1.0800000000000002e-05, | |
| "loss": 1.8512, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.0803579581773354, | |
| "grad_norm": 0.2111833393573761, | |
| "learning_rate": 1.1000000000000001e-05, | |
| "loss": 1.8608, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.08181901196237787, | |
| "grad_norm": 0.24742205440998077, | |
| "learning_rate": 1.1200000000000001e-05, | |
| "loss": 1.9129, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.08328006574742032, | |
| "grad_norm": 0.2268109768629074, | |
| "learning_rate": 1.14e-05, | |
| "loss": 1.7996, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.08474111953246279, | |
| "grad_norm": 0.24065515398979187, | |
| "learning_rate": 1.16e-05, | |
| "loss": 1.9203, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.08620217331750525, | |
| "grad_norm": 0.23937389254570007, | |
| "learning_rate": 1.18e-05, | |
| "loss": 1.8845, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.08766322710254772, | |
| "grad_norm": 0.22344107925891876, | |
| "learning_rate": 1.2e-05, | |
| "loss": 1.94, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.08912428088759017, | |
| "grad_norm": 0.24668040871620178, | |
| "learning_rate": 1.22e-05, | |
| "loss": 1.9473, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.09058533467263263, | |
| "grad_norm": 0.21312135457992554, | |
| "learning_rate": 1.2400000000000002e-05, | |
| "loss": 1.8225, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.0920463884576751, | |
| "grad_norm": 0.2296803593635559, | |
| "learning_rate": 1.2600000000000001e-05, | |
| "loss": 1.7675, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.09350744224271756, | |
| "grad_norm": 0.20073962211608887, | |
| "learning_rate": 1.2800000000000001e-05, | |
| "loss": 1.7844, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.09496849602776002, | |
| "grad_norm": 0.23866504430770874, | |
| "learning_rate": 1.3000000000000001e-05, | |
| "loss": 1.7766, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.09642954981280248, | |
| "grad_norm": 0.24550433456897736, | |
| "learning_rate": 1.3200000000000002e-05, | |
| "loss": 1.9283, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.09789060359784495, | |
| "grad_norm": 0.24073684215545654, | |
| "learning_rate": 1.3400000000000002e-05, | |
| "loss": 1.841, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 0.09935165738288741, | |
| "grad_norm": 0.21318458020687103, | |
| "learning_rate": 1.3600000000000002e-05, | |
| "loss": 1.7465, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.10081271116792986, | |
| "grad_norm": 0.2440112829208374, | |
| "learning_rate": 1.38e-05, | |
| "loss": 1.8503, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 0.10227376495297233, | |
| "grad_norm": 0.21730957925319672, | |
| "learning_rate": 1.4e-05, | |
| "loss": 1.7624, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.1037348187380148, | |
| "grad_norm": 0.22476676106452942, | |
| "learning_rate": 1.4200000000000001e-05, | |
| "loss": 1.8127, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 0.10519587252305726, | |
| "grad_norm": 0.2405499368906021, | |
| "learning_rate": 1.4400000000000001e-05, | |
| "loss": 1.7755, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.10665692630809971, | |
| "grad_norm": 0.24996110796928406, | |
| "learning_rate": 1.46e-05, | |
| "loss": 1.841, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 0.10811798009314218, | |
| "grad_norm": 0.26147621870040894, | |
| "learning_rate": 1.48e-05, | |
| "loss": 1.7769, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 0.10957903387818464, | |
| "grad_norm": 0.23422400653362274, | |
| "learning_rate": 1.5000000000000002e-05, | |
| "loss": 1.7561, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.11104008766322711, | |
| "grad_norm": 0.24707835912704468, | |
| "learning_rate": 1.5200000000000002e-05, | |
| "loss": 1.7038, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.11250114144826956, | |
| "grad_norm": 0.25473567843437195, | |
| "learning_rate": 1.54e-05, | |
| "loss": 1.7496, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 0.11396219523331202, | |
| "grad_norm": 0.3016836941242218, | |
| "learning_rate": 1.5600000000000003e-05, | |
| "loss": 1.7323, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 0.11542324901835449, | |
| "grad_norm": 0.26107528805732727, | |
| "learning_rate": 1.58e-05, | |
| "loss": 1.7511, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 0.11688430280339696, | |
| "grad_norm": 0.3029772639274597, | |
| "learning_rate": 1.6000000000000003e-05, | |
| "loss": 1.691, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.11834535658843941, | |
| "grad_norm": 0.30675315856933594, | |
| "learning_rate": 1.62e-05, | |
| "loss": 1.6696, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 0.11980641037348187, | |
| "grad_norm": 0.3136438727378845, | |
| "learning_rate": 1.64e-05, | |
| "loss": 1.6624, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 0.12126746415852434, | |
| "grad_norm": 0.32071617245674133, | |
| "learning_rate": 1.66e-05, | |
| "loss": 1.6458, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 0.1227285179435668, | |
| "grad_norm": 0.3253510594367981, | |
| "learning_rate": 1.6800000000000002e-05, | |
| "loss": 1.6169, | |
| "step": 84 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 684, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 1, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 9.4533080190677e+16, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |