455 lines
14 KiB
JSON
455 lines
14 KiB
JSON
|
{
|
||
|
|
"best_global_step": null,
|
||
|
|
"best_metric": null,
|
||
|
|
"best_model_checkpoint": null,
|
||
|
|
"epoch": 1.1229813664596273,
|
||
|
|
"eval_steps": 500,
|
||
|
|
"global_step": 302,
|
||
|
|
"is_hyper_param_search": false,
|
||
|
|
"is_local_process_zero": true,
|
||
|
|
"is_world_process_zero": true,
|
||
|
|
"log_history": [
|
||
|
|
{
|
||
|
|
"epoch": 0.037267080745341616,
|
||
|
|
"grad_norm": 0.19933472573757172,
|
||
|
|
"learning_rate": 2.8125000000000003e-05,
|
||
|
|
"loss": 3.1896,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 24.27871,
|
||
|
|
"step": 10,
|
||
|
|
"tokens/total": 368640,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.07,
|
||
|
|
"tokens/trainable": 321256
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.07453416149068323,
|
||
|
|
"grad_norm": 0.2485538274049759,
|
||
|
|
"learning_rate": 4.9995893860576395e-05,
|
||
|
|
"loss": 3.1828,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 24.11418,
|
||
|
|
"step": 20,
|
||
|
|
"tokens/total": 737280,
|
||
|
|
"tokens/train_per_sec_per_gpu": 43.47,
|
||
|
|
"tokens/trainable": 643308
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11180124223602485,
|
||
|
|
"grad_norm": 0.16610555350780487,
|
||
|
|
"learning_rate": 4.99229333433282e-05,
|
||
|
|
"loss": 3.1795,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 24.03473,
|
||
|
|
"step": 30,
|
||
|
|
"tokens/total": 1105920,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.23,
|
||
|
|
"tokens/trainable": 962499
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.14906832298136646,
|
||
|
|
"grad_norm": 0.16600094735622406,
|
||
|
|
"learning_rate": 4.975903174511846e-05,
|
||
|
|
"loss": 3.1002,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.20239,
|
||
|
|
"step": 40,
|
||
|
|
"tokens/total": 1474560,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.06,
|
||
|
|
"tokens/trainable": 1283109
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.18633540372670807,
|
||
|
|
"grad_norm": 0.17242471873760223,
|
||
|
|
"learning_rate": 4.950478712510446e-05,
|
||
|
|
"loss": 3.0775,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.70407,
|
||
|
|
"step": 50,
|
||
|
|
"tokens/total": 1843200,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.42,
|
||
|
|
"tokens/trainable": 1603821
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2236024844720497,
|
||
|
|
"grad_norm": 0.17550165951251984,
|
||
|
|
"learning_rate": 4.9161127194332655e-05,
|
||
|
|
"loss": 3.1334,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.95188,
|
||
|
|
"step": 60,
|
||
|
|
"tokens/total": 2211840,
|
||
|
|
"tokens/train_per_sec_per_gpu": 43.85,
|
||
|
|
"tokens/trainable": 1925471
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2608695652173913,
|
||
|
|
"grad_norm": 0.18663033843040466,
|
||
|
|
"learning_rate": 4.872930593062168e-05,
|
||
|
|
"loss": 3.1937,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 24.37846,
|
||
|
|
"step": 70,
|
||
|
|
"tokens/total": 2580480,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.7,
|
||
|
|
"tokens/trainable": 2242912
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2981366459627329,
|
||
|
|
"grad_norm": 0.20098601281642914,
|
||
|
|
"learning_rate": 4.8210899002933235e-05,
|
||
|
|
"loss": 3.1263,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.7895,
|
||
|
|
"step": 80,
|
||
|
|
"tokens/total": 2949120,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.38,
|
||
|
|
"tokens/trainable": 2562908
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.33540372670807456,
|
||
|
|
"grad_norm": 0.18968884646892548,
|
||
|
|
"learning_rate": 4.760779802192698e-05,
|
||
|
|
"loss": 3.142,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 23.15012,
|
||
|
|
"step": 90,
|
||
|
|
"tokens/total": 3317760,
|
||
|
|
"tokens/train_per_sec_per_gpu": 46.02,
|
||
|
|
"tokens/trainable": 2881497
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.37267080745341613,
|
||
|
|
"grad_norm": 0.21494834125041962,
|
||
|
|
"learning_rate": 4.692220363767833e-05,
|
||
|
|
"loss": 3.1794,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 24.03233,
|
||
|
|
"step": 100,
|
||
|
|
"tokens/total": 3686400,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.95,
|
||
|
|
"tokens/trainable": 3199097
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.40993788819875776,
|
||
|
|
"grad_norm": 0.20388352870941162,
|
||
|
|
"learning_rate": 4.6156617509745015e-05,
|
||
|
|
"loss": 3.1358,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 23.00703,
|
||
|
|
"step": 110,
|
||
|
|
"tokens/total": 4055040,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.96,
|
||
|
|
"tokens/trainable": 3517985
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.4472049689440994,
|
||
|
|
"grad_norm": 0.22123299539089203,
|
||
|
|
"learning_rate": 4.531383317888261e-05,
|
||
|
|
"loss": 3.1032,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.2691,
|
||
|
|
"step": 120,
|
||
|
|
"tokens/total": 4423680,
|
||
|
|
"tokens/train_per_sec_per_gpu": 41.16,
|
||
|
|
"tokens/trainable": 3837324
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.484472049689441,
|
||
|
|
"grad_norm": 0.2072950154542923,
|
||
|
|
"learning_rate": 4.439692587371731e-05,
|
||
|
|
"loss": 3.1074,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.36283,
|
||
|
|
"step": 130,
|
||
|
|
"tokens/total": 4792320,
|
||
|
|
"tokens/train_per_sec_per_gpu": 46.1,
|
||
|
|
"tokens/trainable": 4159485
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.5217391304347826,
|
||
|
|
"grad_norm": 0.22470185160636902,
|
||
|
|
"learning_rate": 4.3409241289570156e-05,
|
||
|
|
"loss": 3.0826,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.81505,
|
||
|
|
"step": 140,
|
||
|
|
"tokens/total": 5160960,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.04,
|
||
|
|
"tokens/trainable": 4479023
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.5590062111801242,
|
||
|
|
"grad_norm": 0.23675978183746338,
|
||
|
|
"learning_rate": 4.235438338037764e-05,
|
||
|
|
"loss": 3.0895,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.96609,
|
||
|
|
"step": 150,
|
||
|
|
"tokens/total": 5529600,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.54,
|
||
|
|
"tokens/trainable": 4798599
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.5962732919254659,
|
||
|
|
"grad_norm": 0.22080311179161072,
|
||
|
|
"learning_rate": 4.123620120825459e-05,
|
||
|
|
"loss": 3.0991,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.17798,
|
||
|
|
"step": 160,
|
||
|
|
"tokens/total": 5898240,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.18,
|
||
|
|
"tokens/trainable": 5117093
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.6335403726708074,
|
||
|
|
"grad_norm": 0.22512708604335785,
|
||
|
|
"learning_rate": 4.005877489868401e-05,
|
||
|
|
"loss": 3.1219,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.68945,
|
||
|
|
"step": 170,
|
||
|
|
"tokens/total": 6266880,
|
||
|
|
"tokens/train_per_sec_per_gpu": 43.85,
|
||
|
|
"tokens/trainable": 5436542
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.6708074534161491,
|
||
|
|
"grad_norm": 0.22725142538547516,
|
||
|
|
"learning_rate": 3.88264007525816e-05,
|
||
|
|
"loss": 3.167,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 23.73617,
|
||
|
|
"step": 180,
|
||
|
|
"tokens/total": 6635520,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.1,
|
||
|
|
"tokens/trainable": 5757152
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.7080745341614907,
|
||
|
|
"grad_norm": 0.23717202246189117,
|
||
|
|
"learning_rate": 3.754357556955966e-05,
|
||
|
|
"loss": 3.1361,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 23.01394,
|
||
|
|
"step": 190,
|
||
|
|
"tokens/total": 7004160,
|
||
|
|
"tokens/train_per_sec_per_gpu": 42.08,
|
||
|
|
"tokens/trainable": 6076610
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.7453416149068323,
|
||
|
|
"grad_norm": 0.2373926043510437,
|
||
|
|
"learning_rate": 3.621498023959294e-05,
|
||
|
|
"loss": 3.1262,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.78722,
|
||
|
|
"step": 200,
|
||
|
|
"tokens/total": 7372800,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.68,
|
||
|
|
"tokens/trainable": 6396405
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.782608695652174,
|
||
|
|
"grad_norm": 0.22468852996826172,
|
||
|
|
"learning_rate": 3.484546266295883e-05,
|
||
|
|
"loss": 3.1138,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.50641,
|
||
|
|
"step": 210,
|
||
|
|
"tokens/total": 7741440,
|
||
|
|
"tokens/train_per_sec_per_gpu": 43.14,
|
||
|
|
"tokens/trainable": 6714000
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.8198757763975155,
|
||
|
|
"grad_norm": 0.23512248694896698,
|
||
|
|
"learning_rate": 3.344002006077487e-05,
|
||
|
|
"loss": 3.113,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.48841,
|
||
|
|
"step": 220,
|
||
|
|
"tokens/total": 8110080,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.86,
|
||
|
|
"tokens/trainable": 7034597
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.8571428571428571,
|
||
|
|
"grad_norm": 0.23428736627101898,
|
||
|
|
"learning_rate": 3.200378074068092e-05,
|
||
|
|
"loss": 3.0454,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.01844,
|
||
|
|
"step": 230,
|
||
|
|
"tokens/total": 8478720,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.72,
|
||
|
|
"tokens/trainable": 7357866
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.8944099378881988,
|
||
|
|
"grad_norm": 0.25676628947257996,
|
||
|
|
"learning_rate": 3.0541985384200686e-05,
|
||
|
|
"loss": 3.1156,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.54695,
|
||
|
|
"step": 240,
|
||
|
|
"tokens/total": 8847360,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.71,
|
||
|
|
"tokens/trainable": 7675991
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.9316770186335404,
|
||
|
|
"grad_norm": 0.2449398934841156,
|
||
|
|
"learning_rate": 2.9059967924063153e-05,
|
||
|
|
"loss": 3.1204,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.65544,
|
||
|
|
"step": 250,
|
||
|
|
"tokens/total": 9216000,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.14,
|
||
|
|
"tokens/trainable": 7995656
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.968944099378882,
|
||
|
|
"grad_norm": 0.2541828453540802,
|
||
|
|
"learning_rate": 2.7563136081260582e-05,
|
||
|
|
"loss": 3.1261,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.78494,
|
||
|
|
"step": 260,
|
||
|
|
"tokens/total": 9584640,
|
||
|
|
"tokens/train_per_sec_per_gpu": 42.8,
|
||
|
|
"tokens/trainable": 8314538
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0037267080745342,
|
||
|
|
"grad_norm": 0.24245300889015198,
|
||
|
|
"learning_rate": 2.6056951632860976e-05,
|
||
|
|
"loss": 3.0867,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.90467,
|
||
|
|
"step": 270,
|
||
|
|
"tokens/total": 9924608,
|
||
|
|
"tokens/train_per_sec_per_gpu": 45.48,
|
||
|
|
"tokens/trainable": 8609040
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0409937888198757,
|
||
|
|
"grad_norm": 0.2593900263309479,
|
||
|
|
"learning_rate": 2.4546910482575908e-05,
|
||
|
|
"loss": 3.0741,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.63041,
|
||
|
|
"step": 280,
|
||
|
|
"tokens/total": 10293248,
|
||
|
|
"tokens/train_per_sec_per_gpu": 44.2,
|
||
|
|
"tokens/trainable": 8927894
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0782608695652174,
|
||
|
|
"grad_norm": 0.2553490996360779,
|
||
|
|
"learning_rate": 2.303852260680388e-05,
|
||
|
|
"loss": 3.0859,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 21.88716,
|
||
|
|
"step": 290,
|
||
|
|
"tokens/total": 10661888,
|
||
|
|
"tokens/train_per_sec_per_gpu": 46.62,
|
||
|
|
"tokens/trainable": 9250881
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.115527950310559,
|
||
|
|
"grad_norm": 0.2889447510242462,
|
||
|
|
"learning_rate": 2.1537291949324007e-05,
|
||
|
|
"loss": 3.0993,
|
||
|
|
"memory/device_reserved (GiB)": 45.16,
|
||
|
|
"memory/max_active (GiB)": 37.89,
|
||
|
|
"memory/max_allocated (GiB)": 37.89,
|
||
|
|
"ppl": 22.18242,
|
||
|
|
"step": 300,
|
||
|
|
"tokens/total": 11030528,
|
||
|
|
"tokens/train_per_sec_per_gpu": 43.46,
|
||
|
|
"tokens/trainable": 9569197
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"logging_steps": 10,
|
||
|
|
"max_steps": 536,
|
||
|
|
"num_input_tokens_seen": 0,
|
||
|
|
"num_train_epochs": 2,
|
||
|
|
"save_steps": 302,
|
||
|
|
"stateful_callbacks": {
|
||
|
|
"TrainerControl": {
|
||
|
|
"args": {
|
||
|
|
"should_epoch_stop": false,
|
||
|
|
"should_evaluate": false,
|
||
|
|
"should_log": false,
|
||
|
|
"should_save": true,
|
||
|
|
"should_training_stop": false
|
||
|
|
},
|
||
|
|
"attributes": {}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"total_flos": 4.7242863209506406e+17,
|
||
|
|
"train_batch_size": 6,
|
||
|
|
"trial_name": null,
|
||
|
|
"trial_params": null
|
||
|
|
}
|