Run 3. Outer Step 10. Inner Step 1.

config.json

@@ -268,17 +268,18 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5421940,
-    5421943,
-    5421947,
-    5421950
+    5422020,
+    5422023,
+    5422026,
+    5422029,
+    5422033
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 0,
+  "inner_step": 1,
   "inner_steps": 0,
   "last_allreduce_block": 5398844,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3696f16251e4cf140b2e7efc156ee8e9ef13117a257cd9db0b37053f02f9f6b0
+oid sha256:1fcd3e4694d4639fac73eac887fd013ec4becb8a03b5a08394663fd8358b652a
 size 8081782026

model.safetensors

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