Run 4. Outer Step 24. Inner Step 35.

config.json

@@ -268,23 +268,25 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5532572,
-    5532574,
-    5532575,
-    5532577,
-    5532579,
-    5532581,
-    5532584,
-    5532586,
-    5532589,
-    5532592
+    5532594,
+    5532596,
+    5532598,
+    5532600,
+    5532601,
+    5532603,
+    5532604,
+    5532606,
+    5532607,
+    5532609,
+    5532611,
+    5532614
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 34,
+  "inner_step": 35,
   "inner_steps": 0,
   "last_allreduce_block": 5529444,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b70d7c236ba1e28cf5c4a77f02811ab334a0904757573a69ccbdf12fd4fb95b1
+oid sha256:008066fae94517468dc1aeab28b3a2fadc6d54f31eab8e9d08654461511ffa5c
 size 8081782026

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:21c1df4c785a1a66e5fa8ece73c470d0637a6eea5e032f25959c2cb4548aec12
+oid sha256:f7c46a768b3024e8861aa450125c837f22ee264a5af6910a4709061f8228076d
 size 4040701744