MHGanainy/gpt2-xl-lora-multi-6

README.md

@@ -15,6 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
 # gpt2-xl-lora-multi-6
 
 This model is a fine-tuned version of [openai-community/gpt2-xl](https://huggingface.co/openai-community/gpt2-xl) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.4480
 
 ## Model description
 

all_results.json

@@ -1,13 +1,13 @@
 {
  "epoch": 1.0,
- "eval_loss": 2.0126099586486816,
- "eval_runtime": 4087.7012,
- "eval_samples_per_second": 7.809,
- "eval_steps_per_second": 0.488,
- "perplexity": 7.482821738733412,
- "total_flos": 1.3078189256852111e+18,
- "train_loss": 2.2911263094506222,
- "train_runtime": 2843.733,
- "train_samples_per_second": 50.495,
- "train_steps_per_second": 3.156
+ "eval_loss": 2.448012590408325,
+ "eval_runtime": 1408.7239,
+ "eval_samples_per_second": 8.454,
+ "eval_steps_per_second": 0.529,
+ "perplexity": 11.56533879845621,
+ "total_flos": 8.45892909419987e+17,
+ "train_loss": 2.683517217184325,
+ "train_runtime": 1637.9851,
+ "train_samples_per_second": 56.701,
+ "train_steps_per_second": 3.544
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
  "epoch": 1.0,
- "eval_loss": 2.0126099586486816,
- "eval_runtime": 4087.7012,
- "eval_samples_per_second": 7.809,
- "eval_steps_per_second": 0.488,
- "perplexity": 7.482821738733412
+ "eval_loss": 2.448012590408325,
+ "eval_runtime": 1408.7239,
+ "eval_samples_per_second": 8.454,
+ "eval_steps_per_second": 0.529,
+ "perplexity": 11.56533879845621
 }

train_results.json

@@ -1,8 +1,8 @@
 {
  "epoch": 1.0,
- "total_flos": 1.3078189256852111e+18,
- "train_loss": 2.2911263094506222,
- "train_runtime": 2843.733,
- "train_samples_per_second": 50.495,
- "train_steps_per_second": 3.156
+ "total_flos": 8.45892909419987e+17,
+ "train_loss": 2.683517217184325,
+ "train_runtime": 1637.9851,
+ "train_samples_per_second": 56.701,
+ "train_steps_per_second": 3.544
 }

trainer_state.json

@@ -3,646 +3,429 @@
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
- "global_step": 8975,
+ "global_step": 5805,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
  {
- "epoch": 0.011142061281337047,
- "grad_norm": 0.04801425710320473,
- "learning_rate": 2.7859033291544785e-07,
- "loss": 2.4677,
+ "epoch": 0.017226528854435832,
+ "grad_norm": 0.04875866696238518,
+ "learning_rate": 4.307559767391773e-07,
+ "loss": 2.816,
  "step": 100
  },
  {
- "epoch": 0.022284122562674095,
- "grad_norm": 0.04515479877591133,
- "learning_rate": 5.571806658308957e-07,
- "loss": 2.465,
+ "epoch": 0.034453057708871665,
+ "grad_norm": 0.04928815737366676,
+ "learning_rate": 8.615119534783546e-07,
+ "loss": 2.8077,
  "step": 200
  },
  {
- "epoch": 0.033426183844011144,
- "grad_norm": 0.04905751347541809,
- "learning_rate": 8.357709987463436e-07,
- "loss": 2.479,
+ "epoch": 0.05167958656330749,
+ "grad_norm": 0.05480952933430672,
+ "learning_rate": 1.292267930217532e-06,
+ "loss": 2.8162,
  "step": 300
  },
  {
- "epoch": 0.04456824512534819,
- "grad_norm": 0.0585104376077652,
- "learning_rate": 1.1143613316617914e-06,
- "loss": 2.4715,
+ "epoch": 0.06890611541774333,
+ "grad_norm": 0.06494510173797607,
+ "learning_rate": 1.7230239069567093e-06,
+ "loss": 2.8237,
  "step": 400
  },
  {
- "epoch": 0.055710306406685235,
- "grad_norm": 0.06588415056467056,
- "learning_rate": 1.3929516645772392e-06,
- "loss": 2.4912,
+ "epoch": 0.08613264427217916,
+ "grad_norm": 0.08336891233921051,
+ "learning_rate": 2.1537798836958863e-06,
+ "loss": 2.7987,
  "step": 500
  },
  {
- "epoch": 0.06685236768802229,
- "grad_norm": 0.07128334790468216,
- "learning_rate": 1.6715419974926873e-06,
- "loss": 2.449,
+ "epoch": 0.10335917312661498,
+ "grad_norm": 0.09097711741924286,
+ "learning_rate": 2.584535860435064e-06,
+ "loss": 2.7894,
  "step": 600
  },
  {
- "epoch": 0.07799442896935933,
- "grad_norm": 0.08448737859725952,
- "learning_rate": 1.9501323304081347e-06,
- "loss": 2.4613,
+ "epoch": 0.12058570198105081,
+ "grad_norm": 0.11103380471467972,
+ "learning_rate": 3.015291837174241e-06,
+ "loss": 2.7909,
  "step": 700
  },
  {
- "epoch": 0.08913649025069638,
- "grad_norm": 0.09213274717330933,
- "learning_rate": 2.2287226633235828e-06,
- "loss": 2.4366,
+ "epoch": 0.13781223083548666,
+ "grad_norm": 0.11578191071748734,
+ "learning_rate": 3.4460478139134186e-06,
+ "loss": 2.7922,
  "step": 800
  },
  {
- "epoch": 0.10027855153203342,
- "grad_norm": 0.10825659334659576,
- "learning_rate": 2.507312996239031e-06,
- "loss": 2.4342,
+ "epoch": 0.15503875968992248,
+ "grad_norm": 0.1395414024591446,
+ "learning_rate": 3.876803790652596e-06,
+ "loss": 2.7704,
  "step": 900
  },
  {
- "epoch": 0.11142061281337047,
- "grad_norm": 0.11989778280258179,
- "learning_rate": 2.7859033291544785e-06,
- "loss": 2.4227,
+ "epoch": 0.17226528854435832,
+ "grad_norm": 0.1293153464794159,
+ "learning_rate": 4.3075597673917726e-06,
+ "loss": 2.7549,
  "step": 1000
  },
  {
- "epoch": 0.12256267409470752,
- "grad_norm": 0.11828930675983429,
- "learning_rate": 3.064493662069926e-06,
- "loss": 2.4346,
+ "epoch": 0.18949181739879414,
+ "grad_norm": 0.13815675675868988,
+ "learning_rate": 4.73831574413095e-06,
+ "loss": 2.7386,
  "step": 1100
  },
  {
- "epoch": 0.13370473537604458,
- "grad_norm": 0.13713237643241882,
- "learning_rate": 3.3430839949853746e-06,
- "loss": 2.3954,
+ "epoch": 0.20671834625322996,
+ "grad_norm": 0.15647341310977936,
+ "learning_rate": 5.169071720870128e-06,
+ "loss": 2.7453,
  "step": 1200
  },
  {
- "epoch": 0.14484679665738162,
- "grad_norm": 0.13783149421215057,
- "learning_rate": 3.6216743279008222e-06,
- "loss": 2.4027,
+ "epoch": 0.2239448751076658,
+ "grad_norm": 0.17875690758228302,
+ "learning_rate": 5.599827697609305e-06,
+ "loss": 2.7357,
  "step": 1300
  },
  {
- "epoch": 0.15598885793871867,
- "grad_norm": 0.14380280673503876,
- "learning_rate": 3.9002646608162694e-06,
- "loss": 2.3894,
+ "epoch": 0.24117140396210163,
+ "grad_norm": 0.16741064190864563,
+ "learning_rate": 6.030583674348482e-06,
+ "loss": 2.739,
  "step": 1400
  },
  {
- "epoch": 0.1671309192200557,
- "grad_norm": 0.1586257964372635,
- "learning_rate": 4.178854993731718e-06,
- "loss": 2.3923,
+ "epoch": 0.25839793281653745,
+ "grad_norm": 0.17629371583461761,
+ "learning_rate": 6.461339651087659e-06,
+ "loss": 2.7257,
  "step": 1500
  },
  {
- "epoch": 0.17827298050139276,
- "grad_norm": 0.15052290260791779,
- "learning_rate": 4.4574453266471655e-06,
- "loss": 2.3792,
+ "epoch": 0.2756244616709733,
+ "grad_norm": 0.18484149873256683,
+ "learning_rate": 6.892095627826837e-06,
+ "loss": 2.712,
  "step": 1600
  },
  {
- "epoch": 0.1894150417827298,
- "grad_norm": 0.16659782826900482,
- "learning_rate": 4.736035659562614e-06,
- "loss": 2.3657,
+ "epoch": 0.29285099052540914,
+ "grad_norm": 0.18326900899410248,
+ "learning_rate": 7.322851604566014e-06,
+ "loss": 2.7207,
  "step": 1700
  },
  {
- "epoch": 0.20055710306406685,
- "grad_norm": 0.16844528913497925,
- "learning_rate": 5.014625992478062e-06,
- "loss": 2.3748,
+ "epoch": 0.31007751937984496,
+ "grad_norm": 0.20282337069511414,
+ "learning_rate": 7.753607581305192e-06,
+ "loss": 2.7264,
  "step": 1800
  },
  {
- "epoch": 0.2116991643454039,
- "grad_norm": 0.17568887770175934,
- "learning_rate": 5.29321632539351e-06,
- "loss": 2.3609,
+ "epoch": 0.3273040482342808,
+ "grad_norm": 0.19705314934253693,
+ "learning_rate": 8.184363558044369e-06,
+ "loss": 2.7107,
  "step": 1900
  },
  {
- "epoch": 0.22284122562674094,
- "grad_norm": 0.19198767840862274,
- "learning_rate": 5.571806658308957e-06,
- "loss": 2.3544,
+ "epoch": 0.34453057708871665,
+ "grad_norm": 0.2224486768245697,
+ "learning_rate": 8.615119534783545e-06,
+ "loss": 2.6999,
  "step": 2000
  },
  {
- "epoch": 0.233983286908078,
- "grad_norm": 0.19439826905727386,
- "learning_rate": 5.850396991224405e-06,
- "loss": 2.3589,
+ "epoch": 0.36175710594315247,
+ "grad_norm": 0.21177536249160767,
+ "learning_rate": 9.045875511522723e-06,
+ "loss": 2.7055,
  "step": 2100
  },
  {
- "epoch": 0.24512534818941503,
- "grad_norm": 0.2022334635257721,
- "learning_rate": 6.128987324139852e-06,
- "loss": 2.3612,
+ "epoch": 0.3789836347975883,
+ "grad_norm": 0.21247607469558716,
+ "learning_rate": 9.4766314882619e-06,
+ "loss": 2.6962,
  "step": 2200
  },
  {
- "epoch": 0.2562674094707521,
- "grad_norm": 0.20067375898361206,
- "learning_rate": 6.4075776570553e-06,
- "loss": 2.3417,
+ "epoch": 0.3962101636520241,
+ "grad_norm": 0.2235407829284668,
+ "learning_rate": 9.907387465001078e-06,
+ "loss": 2.6898,
  "step": 2300
  },
  {
- "epoch": 0.26740947075208915,
- "grad_norm": 0.2165047824382782,
- "learning_rate": 6.686167989970749e-06,
- "loss": 2.3481,
+ "epoch": 0.4134366925064599,
+ "grad_norm": 0.22821404039859772,
+ "learning_rate": 1.0338143441740256e-05,
+ "loss": 2.6896,
  "step": 2400
  },
  {
- "epoch": 0.2785515320334262,
- "grad_norm": 0.21726970374584198,
- "learning_rate": 6.9647583228861955e-06,
- "loss": 2.3331,
+ "epoch": 0.4306632213608958,
+ "grad_norm": 0.21626809239387512,
+ "learning_rate": 1.0768899418479433e-05,
+ "loss": 2.6859,
  "step": 2500
  },
  {
- "epoch": 0.28969359331476324,
- "grad_norm": 0.21808430552482605,
- "learning_rate": 7.2433486558016444e-06,
- "loss": 2.336,
+ "epoch": 0.4478897502153316,
+ "grad_norm": 0.2285173088312149,
+ "learning_rate": 1.119965539521861e-05,
+ "loss": 2.6636,
  "step": 2600
  },
  {
- "epoch": 0.3008356545961003,
- "grad_norm": 0.23211392760276794,
- "learning_rate": 7.5219389887170925e-06,
- "loss": 2.3332,
+ "epoch": 0.46511627906976744,
+ "grad_norm": 0.2364313304424286,
+ "learning_rate": 1.1630411371957786e-05,
+ "loss": 2.6726,
  "step": 2700
  },
  {
- "epoch": 0.31197771587743733,
- "grad_norm": 0.24134476482868195,
- "learning_rate": 7.800529321632539e-06,
- "loss": 2.3377,
+ "epoch": 0.48234280792420325,
+ "grad_norm": 0.2359386384487152,
+ "learning_rate": 1.2061167348696965e-05,
+ "loss": 2.6635,
  "step": 2800
  },
  {
- "epoch": 0.3231197771587744,
- "grad_norm": 0.2478109449148178,
- "learning_rate": 8.079119654547987e-06,
- "loss": 2.3209,
+ "epoch": 0.49956933677863913,
+ "grad_norm": 0.2400447577238083,
+ "learning_rate": 1.2491923325436141e-05,
+ "loss": 2.6934,
  "step": 2900
  },
  {
- "epoch": 0.3342618384401114,
- "grad_norm": 0.25040534138679504,
- "learning_rate": 8.357709987463437e-06,
- "loss": 2.3218,
+ "epoch": 0.5167958656330749,
+ "grad_norm": 0.24956579506397247,
+ "learning_rate": 1.2922679302175318e-05,
+ "loss": 2.6663,
  "step": 3000
  },
  {
- "epoch": 0.34540389972144847,
- "grad_norm": 0.25530460476875305,
- "learning_rate": 8.636300320378883e-06,
- "loss": 2.3063,
+ "epoch": 0.5340223944875108,
+ "grad_norm": 0.24709786474704742,
+ "learning_rate": 1.3353435278914494e-05,
+ "loss": 2.6408,
  "step": 3100
  },
  {
- "epoch": 0.3565459610027855,
- "grad_norm": 0.258489727973938,
- "learning_rate": 8.914890653294331e-06,
- "loss": 2.3218,
+ "epoch": 0.5512489233419466,
+ "grad_norm": 0.2572828233242035,
+ "learning_rate": 1.3784191255653674e-05,
+ "loss": 2.6621,
  "step": 3200
  },
  {
- "epoch": 0.36768802228412256,
- "grad_norm": 0.28399255871772766,
- "learning_rate": 9.193480986209779e-06,
- "loss": 2.3063,
+ "epoch": 0.5684754521963824,
+ "grad_norm": 0.26790568232536316,
+ "learning_rate": 1.421494723239285e-05,
+ "loss": 2.6388,
  "step": 3300
  },
  {
- "epoch": 0.3788300835654596,
- "grad_norm": 0.2863540053367615,
- "learning_rate": 9.472071319125227e-06,
- "loss": 2.2997,
+ "epoch": 0.5857019810508183,
+ "grad_norm": 0.2591148912906647,
+ "learning_rate": 1.4645703209132027e-05,
+ "loss": 2.6415,
  "step": 3400
  },
  {
- "epoch": 0.38997214484679665,
- "grad_norm": 0.2868782877922058,
- "learning_rate": 9.750661652040675e-06,
- "loss": 2.2947,
+ "epoch": 0.602928509905254,
+ "grad_norm": 0.2609754800796509,
+ "learning_rate": 1.5076459185871206e-05,
+ "loss": 2.6588,
  "step": 3500
  },
  {
- "epoch": 0.4011142061281337,
- "grad_norm": 0.2859346270561218,
- "learning_rate": 1.0029251984956123e-05,
- "loss": 2.3054,
+ "epoch": 0.6201550387596899,
+ "grad_norm": 0.2639792859554291,
+ "learning_rate": 1.5507215162610384e-05,
+ "loss": 2.6533,
  "step": 3600
  },
  {
- "epoch": 0.41225626740947074,
- "grad_norm": 0.29307645559310913,
- "learning_rate": 1.030784231787157e-05,
- "loss": 2.2882,
+ "epoch": 0.6373815676141258,
+ "grad_norm": 0.2771030366420746,
+ "learning_rate": 1.593797113934956e-05,
+ "loss": 2.6316,
  "step": 3700
  },
  {
- "epoch": 0.4233983286908078,
- "grad_norm": 0.3006058633327484,
- "learning_rate": 1.058643265078702e-05,
- "loss": 2.2893,
+ "epoch": 0.6546080964685616,
+ "grad_norm": 0.2596473693847656,
+ "learning_rate": 1.6368727116088737e-05,
+ "loss": 2.6474,
  "step": 3800
  },
  {
- "epoch": 0.43454038997214484,
- "grad_norm": 0.31586816906929016,
- "learning_rate": 1.0865022983702467e-05,
- "loss": 2.2967,
+ "epoch": 0.6718346253229974,
+ "grad_norm": 0.2842521369457245,
+ "learning_rate": 1.6799483092827914e-05,
+ "loss": 2.6475,
  "step": 3900
  },
  {
- "epoch": 0.4456824512534819,
- "grad_norm": 0.3375140428543091,
- "learning_rate": 1.1143613316617914e-05,
- "loss": 2.2694,
+ "epoch": 0.6890611541774333,
+ "grad_norm": 0.282993346452713,
+ "learning_rate": 1.723023906956709e-05,
+ "loss": 2.6391,
  "step": 4000
  },
  {
- "epoch": 0.4568245125348189,
- "grad_norm": 0.30628982186317444,
- "learning_rate": 1.142220364953336e-05,
- "loss": 2.2709,
+ "epoch": 0.7062876830318691,
+ "grad_norm": 0.29125678539276123,
+ "learning_rate": 1.766099504630627e-05,
+ "loss": 2.6246,
  "step": 4100
  },
  {
- "epoch": 0.467966573816156,
- "grad_norm": 0.34296101331710815,
- "learning_rate": 1.170079398244881e-05,
- "loss": 2.2783,
+ "epoch": 0.7235142118863049,
+ "grad_norm": 0.2812328338623047,
+ "learning_rate": 1.8091751023045447e-05,
+ "loss": 2.6469,
  "step": 4200
  },
  {
- "epoch": 0.479108635097493,
- "grad_norm": 0.33544307947158813,
- "learning_rate": 1.1979384315364258e-05,
- "loss": 2.2825,
+ "epoch": 0.7407407407407407,
+ "grad_norm": 0.2858037054538727,
+ "learning_rate": 1.8522506999784623e-05,
+ "loss": 2.6348,
  "step": 4300
  },
  {
- "epoch": 0.49025069637883006,
- "grad_norm": 0.3508812487125397,
- "learning_rate": 1.2257974648279704e-05,
- "loss": 2.2793,
+ "epoch": 0.7579672695951766,
+ "grad_norm": 0.30100327730178833,
+ "learning_rate": 1.89532629765238e-05,
+ "loss": 2.6315,
  "step": 4400
  },
  {
- "epoch": 0.5013927576601671,
- "grad_norm": 0.34851330518722534,
- "learning_rate": 1.2536564981195154e-05,
- "loss": 2.2703,
+ "epoch": 0.7751937984496124,
+ "grad_norm": 0.28023988008499146,
+ "learning_rate": 1.9384018953262976e-05,
+ "loss": 2.6244,
  "step": 4500
  },
  {
- "epoch": 0.5125348189415042,
- "grad_norm": 0.3589787185192108,
- "learning_rate": 1.28151553141106e-05,
- "loss": 2.2649,
+ "epoch": 0.7924203273040482,
+ "grad_norm": 0.2947770953178406,
+ "learning_rate": 1.9814774930002156e-05,
+ "loss": 2.6277,
  "step": 4600
  },
  {
- "epoch": 0.5236768802228412,
- "grad_norm": 0.3842375576496124,
- "learning_rate": 1.3093745647026049e-05,
- "loss": 2.264,
+ "epoch": 0.8096468561584841,
+ "grad_norm": 0.30242860317230225,
+ "learning_rate": 1.9881492033537068e-05,
+ "loss": 2.6132,
  "step": 4700
  },
  {
- "epoch": 0.5348189415041783,
- "grad_norm": 0.3963667154312134,
- "learning_rate": 1.3372335979941498e-05,
- "loss": 2.2567,
+ "epoch": 0.8268733850129198,
+ "grad_norm": 0.2965434789657593,
+ "learning_rate": 1.9112585857815527e-05,
+ "loss": 2.6114,
  "step": 4800
  },
  {
- "epoch": 0.5459610027855153,
- "grad_norm": 0.40061014890670776,
- "learning_rate": 1.3650926312856945e-05,
- "loss": 2.2567,
+ "epoch": 0.8440999138673557,
+ "grad_norm": 0.2951013743877411,
+ "learning_rate": 1.7681642907588395e-05,
+ "loss": 2.6349,
  "step": 4900
  },
  {
- "epoch": 0.5571030640668524,
- "grad_norm": 0.3819388747215271,
- "learning_rate": 1.3929516645772391e-05,
- "loss": 2.2545,
+ "epoch": 0.8613264427217916,
+ "grad_norm": 0.34424757957458496,
+ "learning_rate": 1.5692622352080662e-05,
+ "loss": 2.6065,
  "step": 5000
  },
  {
- "epoch": 0.5682451253481894,
- "grad_norm": 0.40177789330482483,
- "learning_rate": 1.420810697868784e-05,
- "loss": 2.2509,
+ "epoch": 0.8785529715762274,
+ "grad_norm": 0.283710241317749,
+ "learning_rate": 1.329002815246645e-05,
+ "loss": 2.6098,
  "step": 5100
  },
  {
- "epoch": 0.5793871866295265,
- "grad_norm": 0.4258297085762024,
- "learning_rate": 1.4486697311603289e-05,
- "loss": 2.2466,
+ "epoch": 0.8957795004306632,
+ "grad_norm": 0.28349170088768005,
+ "learning_rate": 1.064841073155404e-05,
+ "loss": 2.618,
  "step": 5200
  },
  {
- "epoch": 0.5905292479108635,
- "grad_norm": 0.44000041484832764,
- "learning_rate": 1.4765287644518735e-05,
- "loss": 2.2453,
+ "epoch": 0.9130060292850991,
+ "grad_norm": 0.31287553906440735,
+ "learning_rate": 7.95968574359416e-06,
+ "loss": 2.603,
  "step": 5300
  },
  {
- "epoch": 0.6016713091922006,
- "grad_norm": 0.4116784632205963,
- "learning_rate": 1.5043877977434185e-05,
- "loss": 2.2708,
+ "epoch": 0.9302325581395349,
+ "grad_norm": 0.2843669652938843,
+ "learning_rate": 5.419191245899586e-06,
+ "loss": 2.6192,
  "step": 5400
  },
  {
- "epoch": 0.6128133704735376,
- "grad_norm": 0.39489638805389404,
- "learning_rate": 1.5322468310349633e-05,
- "loss": 2.2656,
+ "epoch": 0.9474590869939707,
+ "grad_norm": 0.2816653847694397,
+ "learning_rate": 3.211496230098576e-06,
+ "loss": 2.6142,
  "step": 5500
  },
  {
- "epoch": 0.6239554317548747,
- "grad_norm": 0.43062567710876465,
- "learning_rate": 1.5601058643265078e-05,
- "loss": 2.2305,
+ "epoch": 0.9646856158484065,
+ "grad_norm": 0.30356064438819885,
+ "learning_rate": 1.496991534769564e-06,
+ "loss": 2.6198,
  "step": 5600
  },
  {
- "epoch": 0.6350974930362117,
- "grad_norm": 0.4567050635814667,
- "learning_rate": 1.587964897618053e-05,
- "loss": 2.2392,
+ "epoch": 0.9819121447028424,
+ "grad_norm": 0.27873966097831726,
+ "learning_rate": 4.0023732056077235e-07,
+ "loss": 2.6062,
  "step": 5700
  },
  {
- "epoch": 0.6462395543175488,
- "grad_norm": 0.40272602438926697,
- "learning_rate": 1.6158239309095974e-05,
- "loss": 2.2327,
+ "epoch": 0.9991386735572783,
+ "grad_norm": 0.3346691429615021,
+ "learning_rate": 9.136732052228426e-10,
+ "loss": 2.6072,
  "step": 5800
  },
- {
- "epoch": 0.6573816155988857,
- "grad_norm": 0.4182547628879547,
- "learning_rate": 1.6436829642011422e-05,
- "loss": 2.227,
- "step": 5900
- },
- {
- "epoch": 0.6685236768802229,
- "grad_norm": 0.4616607129573822,
- "learning_rate": 1.6715419974926873e-05,
- "loss": 2.223,
- "step": 6000
- },
- {
- "epoch": 0.6796657381615598,
- "grad_norm": 0.4605095386505127,
- "learning_rate": 1.6994010307842318e-05,
- "loss": 2.229,
- "step": 6100
- },
- {
- "epoch": 0.6908077994428969,
- "grad_norm": 0.48064395785331726,
- "learning_rate": 1.7272600640757766e-05,
- "loss": 2.2285,
- "step": 6200
- },
- {
- "epoch": 0.7019498607242339,
- "grad_norm": 0.45375484228134155,
- "learning_rate": 1.7551190973673214e-05,
- "loss": 2.231,
- "step": 6300
- },
- {
- "epoch": 0.713091922005571,
- "grad_norm": 0.4386723041534424,
- "learning_rate": 1.7829781306588662e-05,
- "loss": 2.2295,
- "step": 6400
- },
- {
- "epoch": 0.724233983286908,
- "grad_norm": 0.4519370198249817,
- "learning_rate": 1.810837163950411e-05,
- "loss": 2.2185,
- "step": 6500
- },
- {
- "epoch": 0.7353760445682451,
- "grad_norm": 0.45770174264907837,
- "learning_rate": 1.8386961972419558e-05,
- "loss": 2.2146,
- "step": 6600
- },
- {
- "epoch": 0.7465181058495822,
- "grad_norm": 0.46970096230506897,
- "learning_rate": 1.8665552305335006e-05,
- "loss": 2.2268,
- "step": 6700
- },
- {
- "epoch": 0.7576601671309192,
- "grad_norm": 0.4301040768623352,
- "learning_rate": 1.8944142638250454e-05,
- "loss": 2.2186,
- "step": 6800
- },
- {
- "epoch": 0.7688022284122563,
- "grad_norm": 0.49275335669517517,
- "learning_rate": 1.9222732971165902e-05,
- "loss": 2.2143,
- "step": 6900
- },
- {
- "epoch": 0.7799442896935933,
- "grad_norm": 0.4531406760215759,
- "learning_rate": 1.950132330408135e-05,
- "loss": 2.2087,
- "step": 7000
- },
- {
- "epoch": 0.7910863509749304,
- "grad_norm": 0.44058841466903687,
- "learning_rate": 1.97799136369968e-05,
- "loss": 2.2081,
- "step": 7100
- },
- {
- "epoch": 0.8022284122562674,
- "grad_norm": 0.4620118737220764,
- "learning_rate": 1.9993253992196043e-05,
- "loss": 2.2029,
- "step": 7200
- },
- {
- "epoch": 0.8133704735376045,
- "grad_norm": 0.4749448299407959,
- "learning_rate": 1.9776845355951603e-05,
- "loss": 2.1974,
- "step": 7300
- },
- {
- "epoch": 0.8245125348189415,
- "grad_norm": 0.4482609033584595,
- "learning_rate": 1.9262050899616325e-05,
- "loss": 2.2229,
- "step": 7400
- },
- {
- "epoch": 0.8356545961002786,
- "grad_norm": 0.4689159095287323,
- "learning_rate": 1.8464581965697866e-05,
- "loss": 2.2081,
- "step": 7500
- },
- {
- "epoch": 0.8467966573816156,
- "grad_norm": 0.4448583126068115,
- "learning_rate": 1.7408777020560473e-05,
- "loss": 2.2181,
- "step": 7600
- },
- {
- "epoch": 0.8579387186629527,
- "grad_norm": 0.513080894947052,
- "learning_rate": 1.6126858853144854e-05,
- "loss": 2.2045,
- "step": 7700
- },
- {
- "epoch": 0.8690807799442897,
- "grad_norm": 0.4850214421749115,
- "learning_rate": 1.465795114698568e-05,
- "loss": 2.1966,
- "step": 7800
- },
- {
- "epoch": 0.8802228412256268,
- "grad_norm": 0.5757237672805786,
- "learning_rate": 1.3046884439396632e-05,
- "loss": 2.2019,
- "step": 7900
- },
- {
- "epoch": 0.8913649025069638,
- "grad_norm": 0.48911547660827637,
- "learning_rate": 1.1342827909521198e-05,
- "loss": 2.1994,
- "step": 8000
- },
- {
- "epoch": 0.9025069637883009,
- "grad_norm": 0.5143416523933411,
- "learning_rate": 9.597788752588108e-06,
- "loss": 2.207,
- "step": 8100
- },
- {
- "epoch": 0.9136490250696379,
- "grad_norm": 0.483064204454422,
- "learning_rate": 7.882117054290375e-06,
- "loss": 2.1971,
- "step": 8200
- },
- {
- "epoch": 0.924791086350975,
- "grad_norm": 0.4610692262649536,
- "learning_rate": 6.213603769467132e-06,
- "loss": 2.1977,
- "step": 8300
- },
- {
- "epoch": 0.935933147632312,
- "grad_norm": 0.50405353307724,
- "learning_rate": 4.660649942045826e-06,
- "loss": 2.1955,
- "step": 8400
- },
- {
- "epoch": 0.947075208913649,
- "grad_norm": 0.47849225997924805,
- "learning_rate": 3.27065116687201e-06,
- "loss": 2.1906,
- "step": 8500
- },
- {
- "epoch": 0.958217270194986,
- "grad_norm": 0.49040549993515015,
- "learning_rate": 2.0860297089355943e-06,
- "loss": 2.1875,
- "step": 8600
- },
- {
- "epoch": 0.9693593314763231,
- "grad_norm": 0.49369877576828003,
- "learning_rate": 1.1429397910307794e-06,
- "loss": 2.2054,
- "step": 8700
- },
- {
- "epoch": 0.9805013927576601,
- "grad_norm": 0.5104972124099731,
- "learning_rate": 4.701641798198353e-07,
- "loss": 2.2121,
- "step": 8800
- },
- {
- "epoch": 0.9916434540389972,
- "grad_norm": 0.48128950595855713,
- "learning_rate": 8.823574609897134e-08,
- "loss": 2.1928,
- "step": 8900
- },
  {
  "epoch": 1.0,
- "step": 8975,
- "total_flos": 1.3078189256852111e+18,
- "train_loss": 2.2911263094506222,
- "train_runtime": 2843.733,
- "train_samples_per_second": 50.495,
- "train_steps_per_second": 3.156
+ "step": 5805,
+ "total_flos": 8.45892909419987e+17,
+ "train_loss": 2.683517217184325,
+ "train_runtime": 1637.9851,
+ "train_samples_per_second": 56.701,
+ "train_steps_per_second": 3.544
  }
  ],
  "logging_steps": 100,
- "max_steps": 8975,
+ "max_steps": 5805,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 500,
@@ -658,7 +441,7 @@
  "attributes": {}
  }
  },
- "total_flos": 1.3078189256852111e+18,
+ "total_flos": 8.45892909419987e+17,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null