Outer Step 2. Inner Step 127. Batch Size 288

config.json

@@ -268,23 +268,23 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5147971,
-    5147974,
-    5147979,
-    5147984,
-    5147989,
-    5147993,
-    5147998,
-    5148002,
-    5148007,
-    5148011
+    5148044,
+    5148048,
+    5148053,
+    5148058,
+    5148063,
+    5148067,
+    5148072,
+    5148076,
+    5148081,
+    5148085
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 118,
+  "inner_step": 127,
   "inner_steps": 0,
   "last_allreduce_block": 5143202,
   "layer_norm_epsilon": 1e-05,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00dddf1d583b80a76d3c3cf9f4cff10613051fe5afa9f74774137e0abf9359c7
+oid sha256:1c1e8a50ac7f2e7c570425a8dedb8e3eff8348170534b0b8c15b73d4f16884e8
 size 8081781770

model.safetensors

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