|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.999854310939306, |
|
"eval_steps": 5000, |
|
"global_step": 3217, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 8.000000000000001e-07, |
|
"loss": 0.1751, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.6333333333333335e-06, |
|
"loss": 0.1343, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 2.3333333333333336e-06, |
|
"loss": 0.1246, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 3.133333333333334e-06, |
|
"loss": 0.1172, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 3.866666666666667e-06, |
|
"loss": 0.1133, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 4.566666666666667e-06, |
|
"loss": 0.1104, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 5.366666666666666e-06, |
|
"loss": 0.1098, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 6.1e-06, |
|
"loss": 0.1077, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 6.833333333333334e-06, |
|
"loss": 0.1061, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 7.566666666666667e-06, |
|
"loss": 0.1068, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 8.3e-06, |
|
"loss": 0.1074, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 9.033333333333334e-06, |
|
"loss": 0.107, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 9.800000000000001e-06, |
|
"loss": 0.1048, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 9.941720946177581e-06, |
|
"loss": 0.1062, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 9.862872814535483e-06, |
|
"loss": 0.1061, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 9.787452862529997e-06, |
|
"loss": 0.1051, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 9.7086047308879e-06, |
|
"loss": 0.1056, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 9.633184778882414e-06, |
|
"loss": 0.1045, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 9.554336647240316e-06, |
|
"loss": 0.1042, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 9.475488515598218e-06, |
|
"loss": 0.105, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 9.39664038395612e-06, |
|
"loss": 0.104, |
|
"step": 525 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 9.321220431950634e-06, |
|
"loss": 0.104, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 9.242372300308537e-06, |
|
"loss": 0.1017, |
|
"step": 575 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 9.166952348303052e-06, |
|
"loss": 0.104, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 9.088104216660953e-06, |
|
"loss": 0.1021, |
|
"step": 625 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 9.012684264655469e-06, |
|
"loss": 0.1027, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 8.940692492286596e-06, |
|
"loss": 0.1041, |
|
"step": 675 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 8.861844360644498e-06, |
|
"loss": 0.1022, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 8.789852588275626e-06, |
|
"loss": 0.1021, |
|
"step": 725 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 8.711004456633528e-06, |
|
"loss": 0.103, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 8.63215632499143e-06, |
|
"loss": 0.1033, |
|
"step": 775 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 8.560164552622558e-06, |
|
"loss": 0.1036, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 8.484744600617072e-06, |
|
"loss": 0.101, |
|
"step": 825 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 8.409324648611588e-06, |
|
"loss": 0.1022, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 8.34076105587933e-06, |
|
"loss": 0.1014, |
|
"step": 875 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 8.265341103873843e-06, |
|
"loss": 0.1005, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 8.186492972231745e-06, |
|
"loss": 0.0999, |
|
"step": 925 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 8.114501199862873e-06, |
|
"loss": 0.0993, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 8.042509427494002e-06, |
|
"loss": 0.1017, |
|
"step": 975 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 7.97051765512513e-06, |
|
"loss": 0.0997, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 7.895097703119644e-06, |
|
"loss": 0.1004, |
|
"step": 1025 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 7.812821391840933e-06, |
|
"loss": 0.1005, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 7.737401439835449e-06, |
|
"loss": 0.0996, |
|
"step": 1075 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 7.661981487829963e-06, |
|
"loss": 0.1, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 7.589989715461091e-06, |
|
"loss": 0.0994, |
|
"step": 1125 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 7.5111415838189925e-06, |
|
"loss": 0.0995, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 7.43914981145012e-06, |
|
"loss": 0.1004, |
|
"step": 1175 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 7.363729859444636e-06, |
|
"loss": 0.1011, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 7.2814535481659246e-06, |
|
"loss": 0.1002, |
|
"step": 1225 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 7.209461775797052e-06, |
|
"loss": 0.099, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 7.137470003428181e-06, |
|
"loss": 0.0996, |
|
"step": 1275 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 7.058621871786082e-06, |
|
"loss": 0.1, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 6.983201919780597e-06, |
|
"loss": 0.0992, |
|
"step": 1325 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 6.904353788138499e-06, |
|
"loss": 0.0993, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 6.825505656496401e-06, |
|
"loss": 0.0975, |
|
"step": 1375 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 6.756942063764142e-06, |
|
"loss": 0.097, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 6.68495029139527e-06, |
|
"loss": 0.0988, |
|
"step": 1425 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 6.612958519026398e-06, |
|
"loss": 0.096, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 6.537538567020913e-06, |
|
"loss": 0.0972, |
|
"step": 1475 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 6.462118615015427e-06, |
|
"loss": 0.0978, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 6.386698663009943e-06, |
|
"loss": 0.0985, |
|
"step": 1525 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 6.3044223517312304e-06, |
|
"loss": 0.0982, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 6.235858758998972e-06, |
|
"loss": 0.098, |
|
"step": 1575 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 6.1638669866301e-06, |
|
"loss": 0.0965, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 6.091875214261228e-06, |
|
"loss": 0.0963, |
|
"step": 1625 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 6.01302708261913e-06, |
|
"loss": 0.097, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 5.937607130613645e-06, |
|
"loss": 0.0972, |
|
"step": 1675 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 5.86218717860816e-06, |
|
"loss": 0.0977, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 5.783339046966061e-06, |
|
"loss": 0.0971, |
|
"step": 1725 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 5.71134727459719e-06, |
|
"loss": 0.0965, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 5.632499142955091e-06, |
|
"loss": 0.0971, |
|
"step": 1775 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 5.557079190949606e-06, |
|
"loss": 0.0967, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 5.485087418580734e-06, |
|
"loss": 0.0949, |
|
"step": 1825 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 5.413095646211862e-06, |
|
"loss": 0.0968, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 5.337675694206377e-06, |
|
"loss": 0.0974, |
|
"step": 1875 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 5.269112101474117e-06, |
|
"loss": 0.0952, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 5.193692149468632e-06, |
|
"loss": 0.0952, |
|
"step": 1925 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 5.118272197463148e-06, |
|
"loss": 0.0967, |
|
"step": 1950 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 5.0462804250942755e-06, |
|
"loss": 0.0948, |
|
"step": 1975 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 4.9708604730887904e-06, |
|
"loss": 0.0967, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 4.892012341446693e-06, |
|
"loss": 0.0953, |
|
"step": 2025 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 4.82002056907782e-06, |
|
"loss": 0.0951, |
|
"step": 2050 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 4.737744257799109e-06, |
|
"loss": 0.0952, |
|
"step": 2075 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 4.662324305793624e-06, |
|
"loss": 0.0962, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 4.586904353788139e-06, |
|
"loss": 0.0953, |
|
"step": 2125 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 4.51834076105588e-06, |
|
"loss": 0.0942, |
|
"step": 2150 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 4.439492629413782e-06, |
|
"loss": 0.0943, |
|
"step": 2175 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 4.360644497771684e-06, |
|
"loss": 0.1057, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 4.288652725402811e-06, |
|
"loss": 0.1048, |
|
"step": 2225 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 4.216660953033939e-06, |
|
"loss": 0.1063, |
|
"step": 2250 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 4.144669180665067e-06, |
|
"loss": 0.1051, |
|
"step": 2275 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 4.069249228659582e-06, |
|
"loss": 0.104, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 3.990401097017484e-06, |
|
"loss": 0.1064, |
|
"step": 2325 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 3.914981145011999e-06, |
|
"loss": 0.1038, |
|
"step": 2350 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 3.84641755227974e-06, |
|
"loss": 0.1038, |
|
"step": 2375 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 3.764141241001029e-06, |
|
"loss": 0.1028, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 3.6887212889955437e-06, |
|
"loss": 0.1044, |
|
"step": 2425 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 3.6167295166266713e-06, |
|
"loss": 0.1052, |
|
"step": 2450 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 3.5413095646211863e-06, |
|
"loss": 0.1042, |
|
"step": 2475 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 3.4693177922523143e-06, |
|
"loss": 0.104, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 3.3973260198834423e-06, |
|
"loss": 0.1025, |
|
"step": 2525 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 3.321906067877957e-06, |
|
"loss": 0.1031, |
|
"step": 2550 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 3.2430579362358587e-06, |
|
"loss": 0.1049, |
|
"step": 2575 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 3.164209804593761e-06, |
|
"loss": 0.1039, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 3.088789852588276e-06, |
|
"loss": 0.1032, |
|
"step": 2625 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 3.0167980802194034e-06, |
|
"loss": 0.1032, |
|
"step": 2650 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 2.9413781282139188e-06, |
|
"loss": 0.1036, |
|
"step": 2675 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 2.8625299965718206e-06, |
|
"loss": 0.1027, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 2.7836818649297224e-06, |
|
"loss": 0.1016, |
|
"step": 2725 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 2.7082619129242378e-06, |
|
"loss": 0.1035, |
|
"step": 2750 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.6294137812821396e-06, |
|
"loss": 0.1025, |
|
"step": 2775 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.557422008913267e-06, |
|
"loss": 0.1013, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.471717517997943e-06, |
|
"loss": 0.1026, |
|
"step": 2825 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 2.3997257456290712e-06, |
|
"loss": 0.1026, |
|
"step": 2850 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 2.331162152896812e-06, |
|
"loss": 0.1019, |
|
"step": 2875 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 2.252314021254714e-06, |
|
"loss": 0.101, |
|
"step": 2900 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 2.183750428522455e-06, |
|
"loss": 0.1026, |
|
"step": 2925 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 2.1083304765169694e-06, |
|
"loss": 0.1014, |
|
"step": 2950 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 2.0363387041480975e-06, |
|
"loss": 0.1022, |
|
"step": 2975 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 1.9574905725059997e-06, |
|
"loss": 0.1015, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 1.8786424408639015e-06, |
|
"loss": 0.1029, |
|
"step": 3025 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 1.8066506684950293e-06, |
|
"loss": 0.1022, |
|
"step": 3050 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 1.7312307164895443e-06, |
|
"loss": 0.1014, |
|
"step": 3075 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 1.6523825848474463e-06, |
|
"loss": 0.1002, |
|
"step": 3100 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.573534453205348e-06, |
|
"loss": 0.1017, |
|
"step": 3125 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 1.504970860473089e-06, |
|
"loss": 0.1015, |
|
"step": 3150 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 1.4226945491943777e-06, |
|
"loss": 0.101, |
|
"step": 3175 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 1.3541309564621189e-06, |
|
"loss": 0.1017, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 3217, |
|
"total_flos": 1.3990082808835046e+22, |
|
"train_loss": 0.006861437764047724, |
|
"train_runtime": 11834.061, |
|
"train_samples_per_second": 348.009, |
|
"train_steps_per_second": 0.272 |
|
} |
|
], |
|
"logging_steps": 25, |
|
"max_steps": 3217, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 200, |
|
"total_flos": 1.3990082808835046e+22, |
|
"train_batch_size": 5, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|