Model save

all_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 1.0,
     "total_flos": 1.997211509414953e+17,
-    "train_loss": 9.503099798246195,
-    "train_runtime": 980.6026,
+    "train_loss": 9.506409932638853,
+    "train_runtime": 976.3941,
     "train_samples": 46801,
-    "train_samples_per_second": 8.538,
+    "train_samples_per_second": 8.574,
     "train_steps_per_second": 0.134
 }

runs/Nov14_21-28-54_main-ah-train-0-0/events.out.tfevents.1731638296.main-ah-train-0-0.519.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6a31fb3127203e45ef17b4cb70d38d1f0ed511d6ce77f951a7faefffecde5df1
-size 12244
+oid sha256:4fed6e185f16728d23e9f82e70ffbfc6d8244487bd95da735e341ee3963ec88f
+size 12869

train_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 1.0,
     "total_flos": 1.997211509414953e+17,
-    "train_loss": 9.503099798246195,
-    "train_runtime": 980.6026,
+    "train_loss": 9.506409932638853,
+    "train_runtime": 976.3941,
     "train_samples": 46801,
-    "train_samples_per_second": 8.538,
+    "train_samples_per_second": 8.574,
     "train_steps_per_second": 0.134
 }

trainer_state.json

@@ -10,208 +10,208 @@
   "log_history": [
     {
       "epoch": 0.007633587786259542,
-      "grad_norm": 177.04818725585938,
+      "grad_norm": 177.04702758789062,
       "learning_rate": 1.4285714285714285e-05,
       "loss": 47.6977,
       "step": 1
     },
     {
       "epoch": 0.03816793893129771,
-      "grad_norm": 85.4582748413086,
+      "grad_norm": 85.58233642578125,
       "learning_rate": 7.142857142857143e-05,
-      "loss": 43.3706,
+      "loss": 43.384,
       "step": 5
     },
     {
       "epoch": 0.07633587786259542,
-      "grad_norm": 21.15643882751465,
+      "grad_norm": 22.778974533081055,
       "learning_rate": 0.00014285714285714287,
-      "loss": 32.4713,
+      "loss": 32.4846,
       "step": 10
     },
     {
       "epoch": 0.11450381679389313,
-      "grad_norm": 12.584202766418457,
+      "grad_norm": 12.44094181060791,
       "learning_rate": 0.00019996395276708856,
-      "loss": 26.3606,
+      "loss": 26.3676,
       "step": 15
     },
     {
       "epoch": 0.15267175572519084,
-      "grad_norm": 6.0736002922058105,
+      "grad_norm": 6.125822067260742,
       "learning_rate": 0.00019870502626379127,
-      "loss": 23.8068,
+      "loss": 23.7984,
       "step": 20
     },
     {
       "epoch": 0.19083969465648856,
-      "grad_norm": 9.369542121887207,
+      "grad_norm": 9.35146427154541,
       "learning_rate": 0.00019566964208274254,
-      "loss": 22.3161,
+      "loss": 22.3143,
       "step": 25
     },
     {
       "epoch": 0.22900763358778625,
-      "grad_norm": 16.93528175354004,
+      "grad_norm": 16.801265716552734,
       "learning_rate": 0.0001909124299802724,
-      "loss": 20.7882,
+      "loss": 20.7786,
       "step": 30
     },
     {
       "epoch": 0.26717557251908397,
-      "grad_norm": 27.67462730407715,
+      "grad_norm": 27.962474822998047,
       "learning_rate": 0.0001845190085543795,
-      "loss": 17.8876,
+      "loss": 17.8884,
       "step": 35
     },
     {
       "epoch": 0.3053435114503817,
-      "grad_norm": 36.653324127197266,
+      "grad_norm": 36.50389099121094,
       "learning_rate": 0.0001766044443118978,
-      "loss": 12.7851,
+      "loss": 12.7762,
       "step": 40
     },
     {
       "epoch": 0.3435114503816794,
-      "grad_norm": 29.700342178344727,
+      "grad_norm": 29.48056983947754,
       "learning_rate": 0.00016731118074275704,
-      "loss": 7.3703,
+      "loss": 7.3657,
       "step": 45
     },
     {
       "epoch": 0.3816793893129771,
-      "grad_norm": 13.149837493896484,
+      "grad_norm": 13.53995418548584,
       "learning_rate": 0.00015680647467311557,
-      "loss": 4.148,
+      "loss": 4.1446,
       "step": 50
     },
     {
       "epoch": 0.4198473282442748,
-      "grad_norm": 9.42947769165039,
+      "grad_norm": 9.89051342010498,
       "learning_rate": 0.00014527938603696376,
-      "loss": 3.4698,
+      "loss": 3.4783,
       "step": 55
     },
     {
       "epoch": 0.4580152671755725,
-      "grad_norm": 4.0911455154418945,
+      "grad_norm": 4.344300270080566,
       "learning_rate": 0.00013293737524320797,
-      "loss": 2.8863,
+      "loss": 2.8938,
       "step": 60
     },
     {
       "epoch": 0.4961832061068702,
-      "grad_norm": 2.124103546142578,
+      "grad_norm": 1.8051707744598389,
       "learning_rate": 0.00012000256937760445,
-      "loss": 2.5943,
+      "loss": 2.6057,
       "step": 65
     },
     {
       "epoch": 0.5343511450381679,
-      "grad_norm": 3.0718891620635986,
+      "grad_norm": 3.4716169834136963,
       "learning_rate": 0.00010670776443910024,
-      "loss": 2.385,
+      "loss": 2.3928,
       "step": 70
     },
     {
       "epoch": 0.5725190839694656,
-      "grad_norm": 3.1116530895233154,
+      "grad_norm": 2.3160088062286377,
       "learning_rate": 9.329223556089975e-05,
-      "loss": 2.2977,
+      "loss": 2.3088,
       "step": 75
     },
     {
       "epoch": 0.6106870229007634,
-      "grad_norm": 0.9759073257446289,
+      "grad_norm": 1.0746971368789673,
       "learning_rate": 7.999743062239557e-05,
-      "loss": 2.1913,
+      "loss": 2.1992,
       "step": 80
     },
     {
       "epoch": 0.648854961832061,
-      "grad_norm": 2.0841591358184814,
+      "grad_norm": 1.997909665107727,
       "learning_rate": 6.706262475679205e-05,
-      "loss": 2.1452,
+      "loss": 2.152,
       "step": 85
     },
     {
       "epoch": 0.6870229007633588,
-      "grad_norm": 1.6919430494308472,
+      "grad_norm": 1.3188209533691406,
       "learning_rate": 5.472061396303629e-05,
-      "loss": 2.1161,
+      "loss": 2.1218,
       "step": 90
     },
     {
       "epoch": 0.7251908396946565,
-      "grad_norm": 0.9739937782287598,
+      "grad_norm": 0.9503675699234009,
       "learning_rate": 4.3193525326884435e-05,
-      "loss": 2.0801,
+      "loss": 2.0813,
       "step": 95
     },
     {
       "epoch": 0.7633587786259542,
-      "grad_norm": 1.5150854587554932,
+      "grad_norm": 1.3065646886825562,
       "learning_rate": 3.268881925724297e-05,
-      "loss": 2.0685,
+      "loss": 2.0714,
       "step": 100
     },
     {
       "epoch": 0.8015267175572519,
-      "grad_norm": 1.2287063598632812,
+      "grad_norm": 1.3419970273971558,
       "learning_rate": 2.339555568810221e-05,
-      "loss": 2.0339,
+      "loss": 2.0359,
       "step": 105
     },
     {
       "epoch": 0.8396946564885496,
-      "grad_norm": 25.910524368286133,
+      "grad_norm": 2.586273670196533,
       "learning_rate": 1.5480991445620542e-05,
-      "loss": 2.0006,
+      "loss": 2.0068,
       "step": 110
     },
     {
       "epoch": 0.8778625954198473,
-      "grad_norm": 0.8792179226875305,
+      "grad_norm": 0.914634108543396,
       "learning_rate": 9.08757001972762e-06,
-      "loss": 2.0391,
+      "loss": 2.0416,
       "step": 115
     },
     {
       "epoch": 0.916030534351145,
-      "grad_norm": 2.187098741531372,
+      "grad_norm": 2.010348081588745,
       "learning_rate": 4.3303579172574885e-06,
-      "loss": 2.0099,
+      "loss": 2.0116,
       "step": 120
     },
     {
       "epoch": 0.9541984732824428,
-      "grad_norm": 1.3658674955368042,
+      "grad_norm": 1.2241454124450684,
       "learning_rate": 1.2949737362087156e-06,
-      "loss": 2.0285,
+      "loss": 2.032,
       "step": 125
     },
     {
       "epoch": 0.9923664122137404,
-      "grad_norm": 1.4813926219940186,
+      "grad_norm": 1.092801809310913,
       "learning_rate": 3.60472329114625e-08,
-      "loss": 2.034,
+      "loss": 2.0377,
       "step": 130
     },
     {
       "epoch": 1.0,
-      "eval_loss": 2.0285837650299072,
-      "eval_runtime": 27.7829,
-      "eval_samples_per_second": 33.15,
-      "eval_steps_per_second": 1.404,
+      "eval_loss": 2.032930850982666,
+      "eval_runtime": 23.8518,
+      "eval_samples_per_second": 38.614,
+      "eval_steps_per_second": 1.216,
       "step": 131
     },
     {
       "epoch": 1.0,
       "step": 131,
       "total_flos": 1.997211509414953e+17,
-      "train_loss": 9.503099798246195,
-      "train_runtime": 980.6026,
-      "train_samples_per_second": 8.538,
+      "train_loss": 9.506409932638853,
+      "train_runtime": 976.3941,
+      "train_samples_per_second": 8.574,
       "train_steps_per_second": 0.134
     }
   ],