|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.6144764723689078, |
|
"eval_steps": 500, |
|
"global_step": 12000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 4.9997567688496474e-05, |
|
"loss": 2.9132, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 4.9990172715142793e-05, |
|
"loss": 0.0721, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 4.997781629993153e-05, |
|
"loss": 0.0757, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 4.9960500896052476e-05, |
|
"loss": 0.0599, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 4.9938477167054135e-05, |
|
"loss": 0.0353, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 4.9911304569050045e-05, |
|
"loss": 0.0256, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.987918618733232e-05, |
|
"loss": 0.0173, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.98421283985469e-05, |
|
"loss": 0.0276, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.9800138559988466e-05, |
|
"loss": 0.0225, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.97532250081397e-05, |
|
"loss": 0.026, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.970139705701628e-05, |
|
"loss": 0.0271, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.9644664996317616e-05, |
|
"loss": 0.0231, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.958304008938407e-05, |
|
"loss": 0.0196, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.951653457096072e-05, |
|
"loss": 0.0124, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.944516164476834e-05, |
|
"loss": 0.0159, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.9368935480882034e-05, |
|
"loss": 0.0144, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.9287871212917866e-05, |
|
"loss": 0.0182, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.9201984935028426e-05, |
|
"loss": 0.018, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.911129369870746e-05, |
|
"loss": 0.0156, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.90158155094046e-05, |
|
"loss": 0.0146, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.8915569322950615e-05, |
|
"loss": 0.0196, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.8810575041794e-05, |
|
"loss": 0.02, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.8700853511049656e-05, |
|
"loss": 0.0143, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.858642651436035e-05, |
|
"loss": 0.0111, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.846731676957191e-05, |
|
"loss": 0.0179, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.834354792422293e-05, |
|
"loss": 0.0133, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.821514455084985e-05, |
|
"loss": 0.0112, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 4.8082132142108465e-05, |
|
"loss": 0.0131, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 4.794453710571272e-05, |
|
"loss": 0.0134, |
|
"step": 2900 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 4.780238675919182e-05, |
|
"loss": 0.0084, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 4.765570932446672e-05, |
|
"loss": 0.0069, |
|
"step": 3100 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 4.75045339222471e-05, |
|
"loss": 0.0186, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 4.734889056624983e-05, |
|
"loss": 0.0138, |
|
"step": 3300 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 4.718881015724017e-05, |
|
"loss": 0.012, |
|
"step": 3400 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 4.702432447689692e-05, |
|
"loss": 0.0121, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 4.6855466181502544e-05, |
|
"loss": 0.0104, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 4.66822687954598e-05, |
|
"loss": 0.014, |
|
"step": 3700 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 4.65047667046359e-05, |
|
"loss": 0.0085, |
|
"step": 3800 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 4.632299514953571e-05, |
|
"loss": 0.0111, |
|
"step": 3900 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 4.613699021830524e-05, |
|
"loss": 0.0122, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 4.59467888395669e-05, |
|
"loss": 0.0082, |
|
"step": 4100 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 4.575242877508777e-05, |
|
"loss": 0.0087, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 4.5553948612282607e-05, |
|
"loss": 0.0115, |
|
"step": 4300 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 4.5351387756552846e-05, |
|
"loss": 0.0069, |
|
"step": 4400 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 4.51447864234632e-05, |
|
"loss": 0.0099, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 4.4934185630757484e-05, |
|
"loss": 0.0087, |
|
"step": 4600 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 4.4719627190215064e-05, |
|
"loss": 0.0096, |
|
"step": 4700 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 4.450115369934976e-05, |
|
"loss": 0.0122, |
|
"step": 4800 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 4.427880853295274e-05, |
|
"loss": 0.0076, |
|
"step": 4900 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 4.4052635834481025e-05, |
|
"loss": 0.0071, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 4.3822680507293455e-05, |
|
"loss": 0.0105, |
|
"step": 5100 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 4.358898820573581e-05, |
|
"loss": 0.0108, |
|
"step": 5200 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 4.3351605326076724e-05, |
|
"loss": 0.0063, |
|
"step": 5300 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 4.3110578997296416e-05, |
|
"loss": 0.0147, |
|
"step": 5400 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 4.286595707172986e-05, |
|
"loss": 0.0135, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 4.261778811556646e-05, |
|
"loss": 0.0105, |
|
"step": 5600 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 4.236612139920786e-05, |
|
"loss": 0.0131, |
|
"step": 5700 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 4.2111006887486035e-05, |
|
"loss": 0.0085, |
|
"step": 5800 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 4.185249522974346e-05, |
|
"loss": 0.0052, |
|
"step": 5900 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 4.159063774977748e-05, |
|
"loss": 0.008, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.1325486435650625e-05, |
|
"loss": 0.011, |
|
"step": 6100 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.105709392936914e-05, |
|
"loss": 0.0074, |
|
"step": 6200 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 4.0785513516431705e-05, |
|
"loss": 0.0147, |
|
"step": 6300 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 4.051079911525031e-05, |
|
"loss": 0.0049, |
|
"step": 6400 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 4.023300526644557e-05, |
|
"loss": 0.0075, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 3.995501009115527e-05, |
|
"loss": 0.0114, |
|
"step": 6600 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 3.967125281105033e-05, |
|
"loss": 0.0089, |
|
"step": 6700 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 3.93845827632495e-05, |
|
"loss": 0.0066, |
|
"step": 6800 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 3.909505686199625e-05, |
|
"loss": 0.0079, |
|
"step": 6900 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 3.880273258852296e-05, |
|
"loss": 0.008, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 3.850766797963886e-05, |
|
"loss": 0.0068, |
|
"step": 7100 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 3.8209921616207645e-05, |
|
"loss": 0.0069, |
|
"step": 7200 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 3.790955261151704e-05, |
|
"loss": 0.0088, |
|
"step": 7300 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 3.7606620599542756e-05, |
|
"loss": 0.0078, |
|
"step": 7400 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"learning_rate": 3.730118572310899e-05, |
|
"loss": 0.008, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 1.02, |
|
"learning_rate": 3.699330862194794e-05, |
|
"loss": 0.0057, |
|
"step": 7600 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"learning_rate": 3.668305042066061e-05, |
|
"loss": 0.0046, |
|
"step": 7700 |
|
}, |
|
{ |
|
"epoch": 1.05, |
|
"learning_rate": 3.637047271658145e-05, |
|
"loss": 0.0059, |
|
"step": 7800 |
|
}, |
|
{ |
|
"epoch": 1.06, |
|
"learning_rate": 3.605563756754904e-05, |
|
"loss": 0.0046, |
|
"step": 7900 |
|
}, |
|
{ |
|
"epoch": 1.08, |
|
"learning_rate": 3.573860747958544e-05, |
|
"loss": 0.0085, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 1.09, |
|
"learning_rate": 3.541944539448648e-05, |
|
"loss": 0.0064, |
|
"step": 8100 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"learning_rate": 3.509821467732553e-05, |
|
"loss": 0.0064, |
|
"step": 8200 |
|
}, |
|
{ |
|
"epoch": 1.12, |
|
"learning_rate": 3.477822117325554e-05, |
|
"loss": 0.0061, |
|
"step": 8300 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 3.445306400520726e-05, |
|
"loss": 0.0087, |
|
"step": 8400 |
|
}, |
|
{ |
|
"epoch": 1.14, |
|
"learning_rate": 3.4126030066319e-05, |
|
"loss": 0.0042, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 1.16, |
|
"learning_rate": 3.379718428450832e-05, |
|
"loss": 0.0071, |
|
"step": 8600 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"learning_rate": 3.346659194740827e-05, |
|
"loss": 0.0043, |
|
"step": 8700 |
|
}, |
|
{ |
|
"epoch": 1.18, |
|
"learning_rate": 3.313431868940551e-05, |
|
"loss": 0.0062, |
|
"step": 8800 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 3.280043047860958e-05, |
|
"loss": 0.0096, |
|
"step": 8900 |
|
}, |
|
{ |
|
"epoch": 1.21, |
|
"learning_rate": 3.24649936037558e-05, |
|
"loss": 0.0049, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"learning_rate": 3.21280746610446e-05, |
|
"loss": 0.0043, |
|
"step": 9100 |
|
}, |
|
{ |
|
"epoch": 1.24, |
|
"learning_rate": 3.1789740540919856e-05, |
|
"loss": 0.006, |
|
"step": 9200 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"learning_rate": 3.145005841478868e-05, |
|
"loss": 0.0053, |
|
"step": 9300 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 3.1109095721685525e-05, |
|
"loss": 0.0075, |
|
"step": 9400 |
|
}, |
|
{ |
|
"epoch": 1.28, |
|
"learning_rate": 3.076692015488315e-05, |
|
"loss": 0.0048, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 1.29, |
|
"learning_rate": 3.0423599648453037e-05, |
|
"loss": 0.0063, |
|
"step": 9600 |
|
}, |
|
{ |
|
"epoch": 1.31, |
|
"learning_rate": 3.007920236377807e-05, |
|
"loss": 0.0075, |
|
"step": 9700 |
|
}, |
|
{ |
|
"epoch": 1.32, |
|
"learning_rate": 2.9733796676020104e-05, |
|
"loss": 0.0082, |
|
"step": 9800 |
|
}, |
|
{ |
|
"epoch": 1.33, |
|
"learning_rate": 2.938745116054501e-05, |
|
"loss": 0.0054, |
|
"step": 9900 |
|
}, |
|
{ |
|
"epoch": 1.35, |
|
"learning_rate": 2.9040234579308025e-05, |
|
"loss": 0.0042, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 1.36, |
|
"learning_rate": 2.8692215867202042e-05, |
|
"loss": 0.0112, |
|
"step": 10100 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"learning_rate": 2.8343464118371603e-05, |
|
"loss": 0.0059, |
|
"step": 10200 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"learning_rate": 2.7994048572495225e-05, |
|
"loss": 0.0034, |
|
"step": 10300 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"learning_rate": 2.7644038601038813e-05, |
|
"loss": 0.0082, |
|
"step": 10400 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 2.7293503693483008e-05, |
|
"loss": 0.0047, |
|
"step": 10500 |
|
}, |
|
{ |
|
"epoch": 1.43, |
|
"learning_rate": 2.6942513443527008e-05, |
|
"loss": 0.0024, |
|
"step": 10600 |
|
}, |
|
{ |
|
"epoch": 1.44, |
|
"learning_rate": 2.6591137535271742e-05, |
|
"loss": 0.0041, |
|
"step": 10700 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"learning_rate": 2.6239445729385104e-05, |
|
"loss": 0.0041, |
|
"step": 10800 |
|
}, |
|
{ |
|
"epoch": 1.47, |
|
"learning_rate": 2.5887507849251914e-05, |
|
"loss": 0.0059, |
|
"step": 10900 |
|
}, |
|
{ |
|
"epoch": 1.48, |
|
"learning_rate": 2.5535393767111533e-05, |
|
"loss": 0.0032, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 1.49, |
|
"learning_rate": 2.5183173390185736e-05, |
|
"loss": 0.004, |
|
"step": 11100 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 2.483091664679959e-05, |
|
"loss": 0.0039, |
|
"step": 11200 |
|
}, |
|
{ |
|
"epoch": 1.52, |
|
"learning_rate": 2.44786934724982e-05, |
|
"loss": 0.004, |
|
"step": 11300 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"learning_rate": 2.4126573796162015e-05, |
|
"loss": 0.0033, |
|
"step": 11400 |
|
}, |
|
{ |
|
"epoch": 1.55, |
|
"learning_rate": 2.377462752612344e-05, |
|
"loss": 0.0091, |
|
"step": 11500 |
|
}, |
|
{ |
|
"epoch": 1.56, |
|
"learning_rate": 2.3422924536287548e-05, |
|
"loss": 0.0049, |
|
"step": 11600 |
|
}, |
|
{ |
|
"epoch": 1.57, |
|
"learning_rate": 2.307153465225958e-05, |
|
"loss": 0.0048, |
|
"step": 11700 |
|
}, |
|
{ |
|
"epoch": 1.59, |
|
"learning_rate": 2.272052763748209e-05, |
|
"loss": 0.0024, |
|
"step": 11800 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"learning_rate": 2.2369973179384445e-05, |
|
"loss": 0.0058, |
|
"step": 11900 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"learning_rate": 2.201994087554733e-05, |
|
"loss": 0.0098, |
|
"step": 12000 |
|
} |
|
], |
|
"logging_steps": 100, |
|
"max_steps": 22296, |
|
"num_train_epochs": 3, |
|
"save_steps": 3000, |
|
"total_flos": 4.8461499181056e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|