Run 4. Outer Step 28. Inner Step 32.

config.json

@@ -268,13 +268,14 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5538123,
-    5538126,
-    5538128,
-    5538131,
-    5538133,
-    5538135,
-    5538138
+    5538141,
+    5538144,
+    5538147,
+    5538149,
+    5538151,
+    5538152,
+    5538153,
+    5538155
   ],
   "block_size": 1024,
   "bos_token_id": 50256,