Run 4. Outer Step 50. Inner Step 1.

config.json

@@ -268,18 +268,25 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5571686,
-    5571687,
-    5571688,
-    5571690,
-    5571691
+    5571694,
+    5571695,
+    5571697,
+    5571698,
+    5571700,
+    5571702,
+    5571704,
+    5571705,
+    5571708,
+    5571710,
+    5571711,
+    5571713
   ],
   "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": 5571611,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a2a0cbd13d1df2fb67e6ed8feaa25aeaee68b7ee9840f2c8914b16ce6f3e484
+oid sha256:596c1096fc49edc964513bf1a3d7435def59478639b32e4ccc996dcc870c232c
 size 8081782026

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:51896991661e02507298ba54890fd1fc262804a915cbfa14522803049877c769
+oid sha256:9594d83826dd7d866c138d748f93fda08e09d9a58e07adab8e2e38be72ee51cf
 size 4040701744