Run 4. Outer Step 46. Inner Step 18.

config.json

@@ -268,43 +268,35 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5567596,
-    5567597,
-    5567598,
-    5567599,
-    5567600,
-    5567601,
-    5567602,
-    5567603,
-    5567604,
-    5567605,
-    5567606,
-    5567607,
-    5567608,
-    5567609,
-    5567611,
-    5567612,
-    5567613,
-    5567613,
-    5567614,
-    5567615,
-    5567615,
-    5567616,
-    5567617,
-    5567617,
-    5567618,
-    5567619,
-    5567620,
-    5567620,
-    5567621,
-    5567622
+    5567622,
+    5567623,
+    5567624,
+    5567625,
+    5567626,
+    5567627,
+    5567628,
+    5567629,
+    5567630,
+    5567631,
+    5567632,
+    5567633,
+    5567634,
+    5567635,
+    5567636,
+    5567637,
+    5567638,
+    5567639,
+    5567640,
+    5567640,
+    5567641,
+    5567642
   ],
   "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": 5566279,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:afa103cbb63652d755057adc425bac80ca33b6299976ccea4cf33f0ab76234e3
+oid sha256:0440c25a947eb4a82294fe28854f3733762f1ac958570f0af8ca287343a97fa8
 size 8081782503

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8f9c79707caece442f204d06e498e5b25501bb998baae822f9f3abd31061dd0c
+oid sha256:a4920d18efe6b3cb1eba443cc5e2fc7ba36e4bca4752b6304603cb78898c49f2
 size 4040701744