Run 4. Outer Step 47. Inner Step 50.

config.json

@@ -268,25 +268,29 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5568473,
-    5568475,
-    5568477,
-    5568479,
-    5568480,
-    5568481,
-    5568483,
-    5568484,
-    5568486,
-    5568487,
-    5568489,
-    5568490
+    5568492,
+    5568494,
+    5568496,
+    5568498,
+    5568499,
+    5568500,
+    5568502,
+    5568503,
+    5568504,
+    5568505,
+    5568506,
+    5568507,
+    5568509,
+    5568509,
+    5568511,
+    5568513
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 49,
+  "inner_step": 50,
   "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:e0ebeeafa96478c37b30c861614befb539ce93f66e4df2fb9a9a3d41bfbc0247
+oid sha256:9841ea3c2c8dd0ddb6647b6dc6536518ac0454b6f79c751b9ffe9277dee3050c
 size 8081782026

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5a5b0621a4b359dd425ddbdff2275cb6bdc92a34f77c5c9c8675cc3db2aba662
+oid sha256:53c177285e36c8ca9b4a295528379fd27dc677a9024aba934b90cb5e17f7f94b
 size 4040701744