Run 4. Outer Step 50. Inner Step 18.

config.json

@@ -268,39 +268,36 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5571844,
-    5571845,
-    5571846,
-    5571847,
-    5571847,
-    5571848,
-    5571849,
-    5571849,
-    5571850,
-    5571851,
-    5571852,
-    5571852,
-    5571852,
-    5571853,
-    5571853,
-    5571854,
-    5571855,
-    5571855,
-    5571856,
-    5571857,
-    5571858,
-    5571859,
-    5571860,
-    5571861,
-    5571862,
-    5571863
+    5571865,
+    5571865,
+    5571866,
+    5571867,
+    5571868,
+    5571869,
+    5571870,
+    5571871,
+    5571872,
+    5571873,
+    5571874,
+    5571874,
+    5571875,
+    5571875,
+    5571876,
+    5571877,
+    5571878,
+    5571879,
+    5571880,
+    5571881,
+    5571882,
+    5571883,
+    5571884
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 16,
+  "inner_step": 18,
   "inner_steps": 0,
   "last_allreduce_block": 5571611,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00dbc58f7ce4b3c011948e75cdda54ffd190206ce8c132b35d4decc79a2f1500
+oid sha256:062f9a9cdf8454997d4fc22de07316d3839579bd5e6d785f6ba8c3b94cecb829
 size 8081782503

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9a39c01e4675ad122d5901e5079770759a2563d38c3a8c26a0d20124c4046a64
+oid sha256:97668fcdcba697c21f3f88d7fc219a26133dfa210c8634714ba8b18f261bb2b3
 size 4040701744