934 lines
21 KiB
JSON
934 lines
21 KiB
JSON
|
{
|
||
|
|
"best_metric": null,
|
||
|
|
"best_model_checkpoint": null,
|
||
|
|
"epoch": 1.9999777777777776,
|
||
|
|
"eval_steps": 500,
|
||
|
|
"global_step": 12857,
|
||
|
|
"is_hyper_param_search": false,
|
||
|
|
"is_local_process_zero": true,
|
||
|
|
"is_world_process_zero": true,
|
||
|
|
"log_history": [
|
||
|
|
{
|
||
|
|
"epoch": 0.02,
|
||
|
|
"grad_norm": 2.191666603088379,
|
||
|
|
"learning_rate": 2.0000000000000002e-07,
|
||
|
|
"loss": 1.3084,
|
||
|
|
"step": 100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.03,
|
||
|
|
"grad_norm": 1.7032761573791504,
|
||
|
|
"learning_rate": 4.0000000000000003e-07,
|
||
|
|
"loss": 1.0978,
|
||
|
|
"step": 200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.05,
|
||
|
|
"grad_norm": 1.4913277626037598,
|
||
|
|
"learning_rate": 6.000000000000001e-07,
|
||
|
|
"loss": 0.8859,
|
||
|
|
"step": 300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06,
|
||
|
|
"grad_norm": 1.470658779144287,
|
||
|
|
"learning_rate": 8.000000000000001e-07,
|
||
|
|
"loss": 0.8098,
|
||
|
|
"step": 400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08,
|
||
|
|
"grad_norm": 1.5057852268218994,
|
||
|
|
"learning_rate": 1.0000000000000002e-06,
|
||
|
|
"loss": 0.7472,
|
||
|
|
"step": 500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09,
|
||
|
|
"grad_norm": 1.3578836917877197,
|
||
|
|
"learning_rate": 1.2000000000000002e-06,
|
||
|
|
"loss": 0.704,
|
||
|
|
"step": 600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11,
|
||
|
|
"grad_norm": 1.3679372072219849,
|
||
|
|
"learning_rate": 1.4000000000000001e-06,
|
||
|
|
"loss": 0.6937,
|
||
|
|
"step": 700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12,
|
||
|
|
"grad_norm": 1.3869675397872925,
|
||
|
|
"learning_rate": 1.6000000000000001e-06,
|
||
|
|
"loss": 0.6667,
|
||
|
|
"step": 800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.14,
|
||
|
|
"grad_norm": 1.209143042564392,
|
||
|
|
"learning_rate": 1.8000000000000001e-06,
|
||
|
|
"loss": 0.6586,
|
||
|
|
"step": 900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.16,
|
||
|
|
"grad_norm": 1.2448453903198242,
|
||
|
|
"learning_rate": 2.0000000000000003e-06,
|
||
|
|
"loss": 0.6464,
|
||
|
|
"step": 1000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.17,
|
||
|
|
"grad_norm": 1.3068097829818726,
|
||
|
|
"learning_rate": 2.2e-06,
|
||
|
|
"loss": 0.6435,
|
||
|
|
"step": 1100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.19,
|
||
|
|
"grad_norm": 1.3215032815933228,
|
||
|
|
"learning_rate": 2.4000000000000003e-06,
|
||
|
|
"loss": 0.6335,
|
||
|
|
"step": 1200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2,
|
||
|
|
"grad_norm": 1.1790049076080322,
|
||
|
|
"learning_rate": 2.6e-06,
|
||
|
|
"loss": 0.6297,
|
||
|
|
"step": 1300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.22,
|
||
|
|
"grad_norm": 1.2376770973205566,
|
||
|
|
"learning_rate": 2.8000000000000003e-06,
|
||
|
|
"loss": 0.6139,
|
||
|
|
"step": 1400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.23,
|
||
|
|
"grad_norm": 1.114210844039917,
|
||
|
|
"learning_rate": 3e-06,
|
||
|
|
"loss": 0.6099,
|
||
|
|
"step": 1500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.25,
|
||
|
|
"grad_norm": 1.1380881071090698,
|
||
|
|
"learning_rate": 3.2000000000000003e-06,
|
||
|
|
"loss": 0.6129,
|
||
|
|
"step": 1600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.26,
|
||
|
|
"grad_norm": 1.2346134185791016,
|
||
|
|
"learning_rate": 3.4000000000000005e-06,
|
||
|
|
"loss": 0.5995,
|
||
|
|
"step": 1700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.28,
|
||
|
|
"grad_norm": 1.293491005897522,
|
||
|
|
"learning_rate": 3.6000000000000003e-06,
|
||
|
|
"loss": 0.5975,
|
||
|
|
"step": 1800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.3,
|
||
|
|
"grad_norm": 1.2162506580352783,
|
||
|
|
"learning_rate": 3.8000000000000005e-06,
|
||
|
|
"loss": 0.5941,
|
||
|
|
"step": 1900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.31,
|
||
|
|
"grad_norm": 1.255366563796997,
|
||
|
|
"learning_rate": 4.000000000000001e-06,
|
||
|
|
"loss": 0.5814,
|
||
|
|
"step": 2000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.33,
|
||
|
|
"grad_norm": 1.230293869972229,
|
||
|
|
"learning_rate": 4.2000000000000004e-06,
|
||
|
|
"loss": 0.5912,
|
||
|
|
"step": 2100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.34,
|
||
|
|
"grad_norm": 1.3100721836090088,
|
||
|
|
"learning_rate": 4.4e-06,
|
||
|
|
"loss": 0.583,
|
||
|
|
"step": 2200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.36,
|
||
|
|
"grad_norm": 1.2346209287643433,
|
||
|
|
"learning_rate": 4.600000000000001e-06,
|
||
|
|
"loss": 0.5716,
|
||
|
|
"step": 2300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.37,
|
||
|
|
"grad_norm": 1.4086166620254517,
|
||
|
|
"learning_rate": 4.800000000000001e-06,
|
||
|
|
"loss": 0.5733,
|
||
|
|
"step": 2400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.39,
|
||
|
|
"grad_norm": 1.1538732051849365,
|
||
|
|
"learning_rate": 5e-06,
|
||
|
|
"loss": 0.5708,
|
||
|
|
"step": 2500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.4,
|
||
|
|
"grad_norm": 1.1694637537002563,
|
||
|
|
"learning_rate": 5.2e-06,
|
||
|
|
"loss": 0.5775,
|
||
|
|
"step": 2600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.42,
|
||
|
|
"grad_norm": 1.1683225631713867,
|
||
|
|
"learning_rate": 5.400000000000001e-06,
|
||
|
|
"loss": 0.5635,
|
||
|
|
"step": 2700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.44,
|
||
|
|
"grad_norm": 1.141072392463684,
|
||
|
|
"learning_rate": 5.600000000000001e-06,
|
||
|
|
"loss": 0.5669,
|
||
|
|
"step": 2800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.45,
|
||
|
|
"grad_norm": 1.0785812139511108,
|
||
|
|
"learning_rate": 5.8e-06,
|
||
|
|
"loss": 0.562,
|
||
|
|
"step": 2900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.47,
|
||
|
|
"grad_norm": 1.0285478830337524,
|
||
|
|
"learning_rate": 6e-06,
|
||
|
|
"loss": 0.5655,
|
||
|
|
"step": 3000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.48,
|
||
|
|
"grad_norm": 1.145011067390442,
|
||
|
|
"learning_rate": 6.200000000000001e-06,
|
||
|
|
"loss": 0.5652,
|
||
|
|
"step": 3100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.5,
|
||
|
|
"grad_norm": 1.1503745317459106,
|
||
|
|
"learning_rate": 6.4000000000000006e-06,
|
||
|
|
"loss": 0.5498,
|
||
|
|
"step": 3200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.51,
|
||
|
|
"grad_norm": 1.0536553859710693,
|
||
|
|
"learning_rate": 6.600000000000001e-06,
|
||
|
|
"loss": 0.5509,
|
||
|
|
"step": 3300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.53,
|
||
|
|
"grad_norm": 1.133914828300476,
|
||
|
|
"learning_rate": 6.800000000000001e-06,
|
||
|
|
"loss": 0.5576,
|
||
|
|
"step": 3400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.54,
|
||
|
|
"grad_norm": 1.0403516292572021,
|
||
|
|
"learning_rate": 7e-06,
|
||
|
|
"loss": 0.5468,
|
||
|
|
"step": 3500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.56,
|
||
|
|
"grad_norm": 1.124706745147705,
|
||
|
|
"learning_rate": 7.2000000000000005e-06,
|
||
|
|
"loss": 0.5481,
|
||
|
|
"step": 3600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.58,
|
||
|
|
"grad_norm": 1.1518924236297607,
|
||
|
|
"learning_rate": 7.4e-06,
|
||
|
|
"loss": 0.5445,
|
||
|
|
"step": 3700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.59,
|
||
|
|
"grad_norm": 1.067154884338379,
|
||
|
|
"learning_rate": 7.600000000000001e-06,
|
||
|
|
"loss": 0.5354,
|
||
|
|
"step": 3800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.61,
|
||
|
|
"grad_norm": 1.0968620777130127,
|
||
|
|
"learning_rate": 7.800000000000002e-06,
|
||
|
|
"loss": 0.5409,
|
||
|
|
"step": 3900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.62,
|
||
|
|
"grad_norm": 1.069420576095581,
|
||
|
|
"learning_rate": 8.000000000000001e-06,
|
||
|
|
"loss": 0.5399,
|
||
|
|
"step": 4000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.64,
|
||
|
|
"grad_norm": 1.1727967262268066,
|
||
|
|
"learning_rate": 8.2e-06,
|
||
|
|
"loss": 0.5398,
|
||
|
|
"step": 4100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.65,
|
||
|
|
"grad_norm": 1.1503219604492188,
|
||
|
|
"learning_rate": 8.400000000000001e-06,
|
||
|
|
"loss": 0.5357,
|
||
|
|
"step": 4200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.67,
|
||
|
|
"grad_norm": 1.1418933868408203,
|
||
|
|
"learning_rate": 8.6e-06,
|
||
|
|
"loss": 0.5308,
|
||
|
|
"step": 4300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.68,
|
||
|
|
"grad_norm": 1.065690517425537,
|
||
|
|
"learning_rate": 8.8e-06,
|
||
|
|
"loss": 0.5405,
|
||
|
|
"step": 4400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.7,
|
||
|
|
"grad_norm": 1.1668462753295898,
|
||
|
|
"learning_rate": 9e-06,
|
||
|
|
"loss": 0.5342,
|
||
|
|
"step": 4500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.72,
|
||
|
|
"grad_norm": 1.1019855737686157,
|
||
|
|
"learning_rate": 9.200000000000002e-06,
|
||
|
|
"loss": 0.5261,
|
||
|
|
"step": 4600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.73,
|
||
|
|
"grad_norm": 1.171284794807434,
|
||
|
|
"learning_rate": 9.4e-06,
|
||
|
|
"loss": 0.5269,
|
||
|
|
"step": 4700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.75,
|
||
|
|
"grad_norm": 1.2499768733978271,
|
||
|
|
"learning_rate": 9.600000000000001e-06,
|
||
|
|
"loss": 0.53,
|
||
|
|
"step": 4800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.76,
|
||
|
|
"grad_norm": 1.0950831174850464,
|
||
|
|
"learning_rate": 9.800000000000001e-06,
|
||
|
|
"loss": 0.5369,
|
||
|
|
"step": 4900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.78,
|
||
|
|
"grad_norm": 1.1700873374938965,
|
||
|
|
"learning_rate": 1e-05,
|
||
|
|
"loss": 0.5296,
|
||
|
|
"step": 5000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.79,
|
||
|
|
"grad_norm": 1.0696507692337036,
|
||
|
|
"learning_rate": 9.963154016212235e-06,
|
||
|
|
"loss": 0.5188,
|
||
|
|
"step": 5100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.81,
|
||
|
|
"grad_norm": 1.1057288646697998,
|
||
|
|
"learning_rate": 9.926308032424467e-06,
|
||
|
|
"loss": 0.5272,
|
||
|
|
"step": 5200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.82,
|
||
|
|
"grad_norm": 0.9978774785995483,
|
||
|
|
"learning_rate": 9.8894620486367e-06,
|
||
|
|
"loss": 0.5288,
|
||
|
|
"step": 5300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.84,
|
||
|
|
"grad_norm": 1.072900652885437,
|
||
|
|
"learning_rate": 9.852616064848932e-06,
|
||
|
|
"loss": 0.5247,
|
||
|
|
"step": 5400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.86,
|
||
|
|
"grad_norm": 0.9971893429756165,
|
||
|
|
"learning_rate": 9.815770081061166e-06,
|
||
|
|
"loss": 0.5137,
|
||
|
|
"step": 5500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.87,
|
||
|
|
"grad_norm": 0.9494544863700867,
|
||
|
|
"learning_rate": 9.778924097273399e-06,
|
||
|
|
"loss": 0.5157,
|
||
|
|
"step": 5600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.89,
|
||
|
|
"grad_norm": 1.04669189453125,
|
||
|
|
"learning_rate": 9.742078113485631e-06,
|
||
|
|
"loss": 0.5267,
|
||
|
|
"step": 5700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.9,
|
||
|
|
"grad_norm": 1.0739444494247437,
|
||
|
|
"learning_rate": 9.705232129697863e-06,
|
||
|
|
"loss": 0.5207,
|
||
|
|
"step": 5800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.92,
|
||
|
|
"grad_norm": 0.9554969668388367,
|
||
|
|
"learning_rate": 9.668386145910098e-06,
|
||
|
|
"loss": 0.5136,
|
||
|
|
"step": 5900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.93,
|
||
|
|
"grad_norm": 0.9823188185691833,
|
||
|
|
"learning_rate": 9.63154016212233e-06,
|
||
|
|
"loss": 0.5137,
|
||
|
|
"step": 6000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.95,
|
||
|
|
"grad_norm": 0.9793609380722046,
|
||
|
|
"learning_rate": 9.594694178334562e-06,
|
||
|
|
"loss": 0.5177,
|
||
|
|
"step": 6100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.96,
|
||
|
|
"grad_norm": 0.9663506746292114,
|
||
|
|
"learning_rate": 9.557848194546795e-06,
|
||
|
|
"loss": 0.5141,
|
||
|
|
"step": 6200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.98,
|
||
|
|
"grad_norm": 1.0234330892562866,
|
||
|
|
"learning_rate": 9.521002210759029e-06,
|
||
|
|
"loss": 0.5077,
|
||
|
|
"step": 6300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0,
|
||
|
|
"grad_norm": 1.2162741422653198,
|
||
|
|
"learning_rate": 9.484156226971261e-06,
|
||
|
|
"loss": 0.5124,
|
||
|
|
"step": 6400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0,
|
||
|
|
"eval_loss": 0.5092498660087585,
|
||
|
|
"eval_runtime": 206.7717,
|
||
|
|
"eval_samples_per_second": 31.184,
|
||
|
|
"eval_steps_per_second": 3.898,
|
||
|
|
"step": 6428
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.01,
|
||
|
|
"grad_norm": 1.1033669710159302,
|
||
|
|
"learning_rate": 9.447310243183494e-06,
|
||
|
|
"loss": 0.4707,
|
||
|
|
"step": 6500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.03,
|
||
|
|
"grad_norm": 1.0372246503829956,
|
||
|
|
"learning_rate": 9.410464259395726e-06,
|
||
|
|
"loss": 0.4532,
|
||
|
|
"step": 6600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.04,
|
||
|
|
"grad_norm": 1.1232597827911377,
|
||
|
|
"learning_rate": 9.37361827560796e-06,
|
||
|
|
"loss": 0.4489,
|
||
|
|
"step": 6700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.06,
|
||
|
|
"grad_norm": 1.061074137687683,
|
||
|
|
"learning_rate": 9.336772291820193e-06,
|
||
|
|
"loss": 0.4552,
|
||
|
|
"step": 6800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.07,
|
||
|
|
"grad_norm": 1.0498615503311157,
|
||
|
|
"learning_rate": 9.299926308032425e-06,
|
||
|
|
"loss": 0.4482,
|
||
|
|
"step": 6900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.09,
|
||
|
|
"grad_norm": 1.0324976444244385,
|
||
|
|
"learning_rate": 9.263080324244657e-06,
|
||
|
|
"loss": 0.4511,
|
||
|
|
"step": 7000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.1,
|
||
|
|
"grad_norm": 0.9778733849525452,
|
||
|
|
"learning_rate": 9.226234340456892e-06,
|
||
|
|
"loss": 0.4452,
|
||
|
|
"step": 7100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.12,
|
||
|
|
"grad_norm": 1.0444563627243042,
|
||
|
|
"learning_rate": 9.189388356669124e-06,
|
||
|
|
"loss": 0.4512,
|
||
|
|
"step": 7200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.14,
|
||
|
|
"grad_norm": 1.0546364784240723,
|
||
|
|
"learning_rate": 9.152542372881356e-06,
|
||
|
|
"loss": 0.4486,
|
||
|
|
"step": 7300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.15,
|
||
|
|
"grad_norm": 1.0815290212631226,
|
||
|
|
"learning_rate": 9.115696389093589e-06,
|
||
|
|
"loss": 0.4514,
|
||
|
|
"step": 7400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.17,
|
||
|
|
"grad_norm": 1.0215433835983276,
|
||
|
|
"learning_rate": 9.078850405305823e-06,
|
||
|
|
"loss": 0.4467,
|
||
|
|
"step": 7500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.18,
|
||
|
|
"grad_norm": 0.9386591911315918,
|
||
|
|
"learning_rate": 9.042004421518055e-06,
|
||
|
|
"loss": 0.4515,
|
||
|
|
"step": 7600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.2,
|
||
|
|
"grad_norm": 1.08309006690979,
|
||
|
|
"learning_rate": 9.005158437730288e-06,
|
||
|
|
"loss": 0.4466,
|
||
|
|
"step": 7700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.21,
|
||
|
|
"grad_norm": 1.0043574571609497,
|
||
|
|
"learning_rate": 8.968312453942522e-06,
|
||
|
|
"loss": 0.451,
|
||
|
|
"step": 7800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.23,
|
||
|
|
"grad_norm": 0.9670729041099548,
|
||
|
|
"learning_rate": 8.931466470154754e-06,
|
||
|
|
"loss": 0.4508,
|
||
|
|
"step": 7900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.24,
|
||
|
|
"grad_norm": 1.001947283744812,
|
||
|
|
"learning_rate": 8.894620486366987e-06,
|
||
|
|
"loss": 0.4454,
|
||
|
|
"step": 8000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.26,
|
||
|
|
"grad_norm": 0.9632651209831238,
|
||
|
|
"learning_rate": 8.857774502579219e-06,
|
||
|
|
"loss": 0.4419,
|
||
|
|
"step": 8100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.28,
|
||
|
|
"grad_norm": 1.0596399307250977,
|
||
|
|
"learning_rate": 8.820928518791453e-06,
|
||
|
|
"loss": 0.444,
|
||
|
|
"step": 8200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.29,
|
||
|
|
"grad_norm": 1.0315296649932861,
|
||
|
|
"learning_rate": 8.784082535003686e-06,
|
||
|
|
"loss": 0.4443,
|
||
|
|
"step": 8300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.31,
|
||
|
|
"grad_norm": 1.2347131967544556,
|
||
|
|
"learning_rate": 8.747236551215918e-06,
|
||
|
|
"loss": 0.4478,
|
||
|
|
"step": 8400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.32,
|
||
|
|
"grad_norm": 1.1402790546417236,
|
||
|
|
"learning_rate": 8.71039056742815e-06,
|
||
|
|
"loss": 0.4458,
|
||
|
|
"step": 8500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.34,
|
||
|
|
"grad_norm": 1.0721864700317383,
|
||
|
|
"learning_rate": 8.673544583640385e-06,
|
||
|
|
"loss": 0.4391,
|
||
|
|
"step": 8600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.35,
|
||
|
|
"grad_norm": 1.072485327720642,
|
||
|
|
"learning_rate": 8.636698599852617e-06,
|
||
|
|
"loss": 0.4443,
|
||
|
|
"step": 8700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.37,
|
||
|
|
"grad_norm": 1.0993961095809937,
|
||
|
|
"learning_rate": 8.59985261606485e-06,
|
||
|
|
"loss": 0.4485,
|
||
|
|
"step": 8800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.38,
|
||
|
|
"grad_norm": 1.1231122016906738,
|
||
|
|
"learning_rate": 8.563006632277082e-06,
|
||
|
|
"loss": 0.4423,
|
||
|
|
"step": 8900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.4,
|
||
|
|
"grad_norm": 1.131405234336853,
|
||
|
|
"learning_rate": 8.526160648489316e-06,
|
||
|
|
"loss": 0.4512,
|
||
|
|
"step": 9000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.42,
|
||
|
|
"grad_norm": 1.0044902563095093,
|
||
|
|
"learning_rate": 8.489314664701548e-06,
|
||
|
|
"loss": 0.444,
|
||
|
|
"step": 9100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.43,
|
||
|
|
"grad_norm": 0.9530848860740662,
|
||
|
|
"learning_rate": 8.45246868091378e-06,
|
||
|
|
"loss": 0.4423,
|
||
|
|
"step": 9200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.45,
|
||
|
|
"grad_norm": 0.8441118597984314,
|
||
|
|
"learning_rate": 8.415622697126013e-06,
|
||
|
|
"loss": 0.4443,
|
||
|
|
"step": 9300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.46,
|
||
|
|
"grad_norm": 0.9523534774780273,
|
||
|
|
"learning_rate": 8.378776713338247e-06,
|
||
|
|
"loss": 0.4403,
|
||
|
|
"step": 9400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.48,
|
||
|
|
"grad_norm": 1.0068507194519043,
|
||
|
|
"learning_rate": 8.34193072955048e-06,
|
||
|
|
"loss": 0.4364,
|
||
|
|
"step": 9500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.49,
|
||
|
|
"grad_norm": 1.0632400512695312,
|
||
|
|
"learning_rate": 8.305084745762712e-06,
|
||
|
|
"loss": 0.4414,
|
||
|
|
"step": 9600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.51,
|
||
|
|
"grad_norm": 1.1296297311782837,
|
||
|
|
"learning_rate": 8.268238761974944e-06,
|
||
|
|
"loss": 0.4429,
|
||
|
|
"step": 9700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.52,
|
||
|
|
"grad_norm": 1.0164484977722168,
|
||
|
|
"learning_rate": 8.231392778187179e-06,
|
||
|
|
"loss": 0.4403,
|
||
|
|
"step": 9800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.54,
|
||
|
|
"grad_norm": 1.1088131666183472,
|
||
|
|
"learning_rate": 8.194546794399411e-06,
|
||
|
|
"loss": 0.4407,
|
||
|
|
"step": 9900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.56,
|
||
|
|
"grad_norm": 1.041729211807251,
|
||
|
|
"learning_rate": 8.157700810611643e-06,
|
||
|
|
"loss": 0.4434,
|
||
|
|
"step": 10000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.57,
|
||
|
|
"grad_norm": 1.0203355550765991,
|
||
|
|
"learning_rate": 8.120854826823877e-06,
|
||
|
|
"loss": 0.4428,
|
||
|
|
"step": 10100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.59,
|
||
|
|
"grad_norm": 1.0350580215454102,
|
||
|
|
"learning_rate": 8.08400884303611e-06,
|
||
|
|
"loss": 0.4365,
|
||
|
|
"step": 10200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.6,
|
||
|
|
"grad_norm": 1.1208500862121582,
|
||
|
|
"learning_rate": 8.047162859248342e-06,
|
||
|
|
"loss": 0.4426,
|
||
|
|
"step": 10300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.62,
|
||
|
|
"grad_norm": 1.1885343790054321,
|
||
|
|
"learning_rate": 8.010316875460575e-06,
|
||
|
|
"loss": 0.4414,
|
||
|
|
"step": 10400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.63,
|
||
|
|
"grad_norm": 0.9988847374916077,
|
||
|
|
"learning_rate": 7.973470891672809e-06,
|
||
|
|
"loss": 0.4368,
|
||
|
|
"step": 10500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.65,
|
||
|
|
"grad_norm": 1.0181776285171509,
|
||
|
|
"learning_rate": 7.936624907885041e-06,
|
||
|
|
"loss": 0.4399,
|
||
|
|
"step": 10600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.66,
|
||
|
|
"grad_norm": 0.9755112528800964,
|
||
|
|
"learning_rate": 7.899778924097274e-06,
|
||
|
|
"loss": 0.4424,
|
||
|
|
"step": 10700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.68,
|
||
|
|
"grad_norm": 1.0309982299804688,
|
||
|
|
"learning_rate": 7.862932940309506e-06,
|
||
|
|
"loss": 0.4418,
|
||
|
|
"step": 10800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.7,
|
||
|
|
"grad_norm": 1.032485842704773,
|
||
|
|
"learning_rate": 7.82608695652174e-06,
|
||
|
|
"loss": 0.4404,
|
||
|
|
"step": 10900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.71,
|
||
|
|
"grad_norm": 1.0489119291305542,
|
||
|
|
"learning_rate": 7.789240972733973e-06,
|
||
|
|
"loss": 0.4333,
|
||
|
|
"step": 11000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.73,
|
||
|
|
"grad_norm": 1.1144189834594727,
|
||
|
|
"learning_rate": 7.752394988946205e-06,
|
||
|
|
"loss": 0.4348,
|
||
|
|
"step": 11100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.74,
|
||
|
|
"grad_norm": 1.0886303186416626,
|
||
|
|
"learning_rate": 7.715549005158437e-06,
|
||
|
|
"loss": 0.4355,
|
||
|
|
"step": 11200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.76,
|
||
|
|
"grad_norm": 1.0355674028396606,
|
||
|
|
"learning_rate": 7.678703021370671e-06,
|
||
|
|
"loss": 0.4348,
|
||
|
|
"step": 11300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.77,
|
||
|
|
"grad_norm": 1.0847941637039185,
|
||
|
|
"learning_rate": 7.641857037582904e-06,
|
||
|
|
"loss": 0.4398,
|
||
|
|
"step": 11400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.79,
|
||
|
|
"grad_norm": 1.0430936813354492,
|
||
|
|
"learning_rate": 7.605011053795137e-06,
|
||
|
|
"loss": 0.4316,
|
||
|
|
"step": 11500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.8,
|
||
|
|
"grad_norm": 1.0820753574371338,
|
||
|
|
"learning_rate": 7.5681650700073696e-06,
|
||
|
|
"loss": 0.431,
|
||
|
|
"step": 11600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.82,
|
||
|
|
"grad_norm": 1.080796241760254,
|
||
|
|
"learning_rate": 7.531319086219603e-06,
|
||
|
|
"loss": 0.433,
|
||
|
|
"step": 11700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.84,
|
||
|
|
"grad_norm": 1.01823091506958,
|
||
|
|
"learning_rate": 7.494473102431835e-06,
|
||
|
|
"loss": 0.435,
|
||
|
|
"step": 11800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.85,
|
||
|
|
"grad_norm": 1.0962271690368652,
|
||
|
|
"learning_rate": 7.4576271186440685e-06,
|
||
|
|
"loss": 0.4347,
|
||
|
|
"step": 11900
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.87,
|
||
|
|
"grad_norm": 1.0160537958145142,
|
||
|
|
"learning_rate": 7.420781134856301e-06,
|
||
|
|
"loss": 0.4326,
|
||
|
|
"step": 12000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.88,
|
||
|
|
"grad_norm": 0.9727386832237244,
|
||
|
|
"learning_rate": 7.383935151068534e-06,
|
||
|
|
"loss": 0.4329,
|
||
|
|
"step": 12100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.9,
|
||
|
|
"grad_norm": 1.0276840925216675,
|
||
|
|
"learning_rate": 7.3470891672807666e-06,
|
||
|
|
"loss": 0.4295,
|
||
|
|
"step": 12200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.91,
|
||
|
|
"grad_norm": 0.972959041595459,
|
||
|
|
"learning_rate": 7.310243183493e-06,
|
||
|
|
"loss": 0.4252,
|
||
|
|
"step": 12300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.93,
|
||
|
|
"grad_norm": 1.0518746376037598,
|
||
|
|
"learning_rate": 7.273397199705232e-06,
|
||
|
|
"loss": 0.4299,
|
||
|
|
"step": 12400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.94,
|
||
|
|
"grad_norm": 1.0917232036590576,
|
||
|
|
"learning_rate": 7.2365512159174655e-06,
|
||
|
|
"loss": 0.4326,
|
||
|
|
"step": 12500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.96,
|
||
|
|
"grad_norm": 1.0608563423156738,
|
||
|
|
"learning_rate": 7.199705232129699e-06,
|
||
|
|
"loss": 0.4318,
|
||
|
|
"step": 12600
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.98,
|
||
|
|
"grad_norm": 1.0238438844680786,
|
||
|
|
"learning_rate": 7.162859248341931e-06,
|
||
|
|
"loss": 0.4298,
|
||
|
|
"step": 12700
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.99,
|
||
|
|
"grad_norm": 0.9618347883224487,
|
||
|
|
"learning_rate": 7.126013264554164e-06,
|
||
|
|
"loss": 0.4351,
|
||
|
|
"step": 12800
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 2.0,
|
||
|
|
"eval_loss": 0.47487419843673706,
|
||
|
|
"eval_runtime": 205.6987,
|
||
|
|
"eval_samples_per_second": 31.347,
|
||
|
|
"eval_steps_per_second": 3.918,
|
||
|
|
"step": 12857
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"logging_steps": 100,
|
||
|
|
"max_steps": 32140,
|
||
|
|
"num_input_tokens_seen": 0,
|
||
|
|
"num_train_epochs": 5,
|
||
|
|
"save_steps": 500,
|
||
|
|
"total_flos": 6.404792140239882e+17,
|
||
|
|
"train_batch_size": 2,
|
||
|
|
"trial_name": null,
|
||
|
|
"trial_params": null
|
||
|
|
}
|