Run 4. Outer Step 29. Inner Step 20.

config.json

@@ -268,18 +268,25 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5539125,
-    5539126,
-    5539128,
-    5539131,
-    5539133
+    5539135,
+    5539139,
+    5539141,
+    5539144,
+    5539146,
+    5539148,
+    5539150,
+    5539151,
+    5539151,
+    5539154,
+    5539156,
+    5539158
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 19,
+  "inner_step": 20,
   "inner_steps": 0,
   "last_allreduce_block": 5538408,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eea3ca636c43cbaec5f22dd7216566252d94f66e5b14d31239b02b33744f6aff
+oid sha256:03ea83ce85b07f7e4a5e306e39a54efb87ad879441e1ea91faa60882e0c606c6
 size 8081782026

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2598a76244f8e870a79f0d8274464aebb83a6444a03d0a2dbf8eb34465c5b079
+oid sha256:d7ceb12d107e280eba2b44331df18f1416d016afd47e8661a04650ffa854e1e5
 size 4040701744