Run 4. Outer Step 56. Inner Step 10.

config.json

@@ -267,13 +267,19 @@
     "AutoConfig": "distributed/optimized-gpt2-500m--configuration_gpt_optimized.GPTOptimConfig",
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
-  "block_list": [],
+  "block_list": [
+    5599797,
+    5599801,
+    5599805,
+    5599810,
+    5599814
+  ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 8,
+  "inner_step": 10,
   "inner_steps": 0,
   "last_allreduce_block": 5576204,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7aaf7adfb5b15185ce8e81267fd8d03223c38ada2ab6b8b9600ccbb99e18b92a
+oid sha256:46f1c472fa3c87dceb446052db740f1782e1c9b32b208c401c50a128e23e3d5e
 size 8081782026

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:300545904e3bfa7e344293cf33d69628dcdf9a13b25432d1c6c99173fcf44134
+oid sha256:694f0e05d768bbe31b6f85db33b4a727aa3cdaf906a31f6a862634ca3c3440ef
 size 4040701744