Outer Step 3. Inner Step 5. Batch Size 404

config.json

@@ -3,50 +3,50 @@
   "activation_function": "gelu_new",
   "all_reduce_scores": {
     "0": "NON_PARTICIPATING",
-    "1": "SUCCESS",
+    "1": "NON_PARTICIPATING",
     "10": "NON_PARTICIPATING",
     "100": "NON_PARTICIPATING",
     "101": "NON_PARTICIPATING",
     "102": "NON_PARTICIPATING",
     "103": "NON_PARTICIPATING",
     "104": "NON_PARTICIPATING",
-    "105": "SUCCESS",
-    "106": "SUCCESS",
+    "105": "NON_PARTICIPATING",
+    "106": "NON_PARTICIPATING",
     "107": "NON_PARTICIPATING",
     "108": "NON_PARTICIPATING",
     "109": "NON_PARTICIPATING",
     "11": "NON_PARTICIPATING",
-    "110": "FAIL",
+    "110": "NON_PARTICIPATING",
     "111": "NON_PARTICIPATING",
     "112": "NON_PARTICIPATING",
     "113": "NON_PARTICIPATING",
     "114": "NON_PARTICIPATING",
-    "115": "FAIL",
+    "115": "NON_PARTICIPATING",
     "116": "NON_PARTICIPATING",
-    "117": "FAIL",
+    "117": "NON_PARTICIPATING",
     "118": "NON_PARTICIPATING",
     "119": "NON_PARTICIPATING",
     "12": "NON_PARTICIPATING",
     "120": "NON_PARTICIPATING",
-    "121": "SUCCESS",
-    "122": "SUCCESS",
+    "121": "NON_PARTICIPATING",
+    "122": "NON_PARTICIPATING",
     "123": "NON_PARTICIPATING",
     "124": "NON_PARTICIPATING",
     "125": "NON_PARTICIPATING",
-    "126": "SUCCESS",
-    "127": "SUCCESS",
+    "126": "NON_PARTICIPATING",
+    "127": "NON_PARTICIPATING",
     "128": "NON_PARTICIPATING",
-    "129": "FAIL",
+    "129": "NON_PARTICIPATING",
     "13": "NON_PARTICIPATING",
-    "130": "SUCCESS",
+    "130": "NON_PARTICIPATING",
     "131": "NON_PARTICIPATING",
     "132": "NON_PARTICIPATING",
     "133": "NON_PARTICIPATING",
     "134": "NON_PARTICIPATING",
     "135": "NON_PARTICIPATING",
     "136": "NON_PARTICIPATING",
-    "137": "SUCCESS",
-    "138": "SUCCESS",
+    "137": "NON_PARTICIPATING",
+    "138": "NON_PARTICIPATING",
     "139": "NON_PARTICIPATING",
     "14": "NON_PARTICIPATING",
     "140": "NON_PARTICIPATING",
@@ -54,126 +54,126 @@
     "142": "NON_PARTICIPATING",
     "143": "NON_PARTICIPATING",
     "144": "NON_PARTICIPATING",
-    "145": "SUCCESS",
-    "146": "FAIL",
+    "145": "NON_PARTICIPATING",
+    "146": "NON_PARTICIPATING",
     "147": "NON_PARTICIPATING",
-    "148": "FAIL",
+    "148": "NON_PARTICIPATING",
     "149": "NON_PARTICIPATING",
-    "15": "FAIL",
+    "15": "NON_PARTICIPATING",
     "150": "NON_PARTICIPATING",
     "151": "NON_PARTICIPATING",
     "152": "NON_PARTICIPATING",
     "153": "NON_PARTICIPATING",
-    "154": "SUCCESS",
+    "154": "NON_PARTICIPATING",
     "155": "NON_PARTICIPATING",
     "156": "NON_PARTICIPATING",
     "157": "NON_PARTICIPATING",
-    "158": "SUCCESS",
-    "159": "SUCCESS",
+    "158": "NON_PARTICIPATING",
+    "159": "NON_PARTICIPATING",
     "16": "NON_PARTICIPATING",
-    "160": "SUCCESS",
+    "160": "NON_PARTICIPATING",
     "161": "NON_PARTICIPATING",
     "162": "NON_PARTICIPATING",
     "163": "NON_PARTICIPATING",
-    "164": "FAIL",
+    "164": "NON_PARTICIPATING",
     "165": "NON_PARTICIPATING",
-    "166": "SUCCESS",
+    "166": "NON_PARTICIPATING",
     "167": "NON_PARTICIPATING",
-    "168": "SUCCESS",
+    "168": "NON_PARTICIPATING",
     "169": "NON_PARTICIPATING",
     "17": "NON_PARTICIPATING",
-    "170": "FAIL",
-    "171": "SUCCESS",
+    "170": "NON_PARTICIPATING",
+    "171": "NON_PARTICIPATING",
     "172": "NON_PARTICIPATING",
     "173": "NON_PARTICIPATING",
-    "174": "SUCCESS",
+    "174": "NON_PARTICIPATING",
     "175": "NON_PARTICIPATING",
     "176": "NON_PARTICIPATING",
     "177": "NON_PARTICIPATING",
-    "178": "FAIL",
+    "178": "NON_PARTICIPATING",
     "179": "NON_PARTICIPATING",
     "18": "NON_PARTICIPATING",
-    "180": "FAIL",
+    "180": "NON_PARTICIPATING",
     "181": "NON_PARTICIPATING",
     "182": "NON_PARTICIPATING",
     "183": "NON_PARTICIPATING",
     "184": "NON_PARTICIPATING",
     "185": "NON_PARTICIPATING",
-    "186": "SUCCESS",
+    "186": "NON_PARTICIPATING",
     "187": "NON_PARTICIPATING",
     "188": "NON_PARTICIPATING",
-    "189": "SUCCESS",
+    "189": "NON_PARTICIPATING",
     "19": "NON_PARTICIPATING",
-    "190": "FAIL",
+    "190": "NON_PARTICIPATING",
     "191": "NON_PARTICIPATING",
-    "192": "SUCCESS",
+    "192": "NON_PARTICIPATING",
     "193": "NON_PARTICIPATING",
     "194": "NON_PARTICIPATING",
-    "195": "SUCCESS",
-    "196": "SUCCESS",
+    "195": "NON_PARTICIPATING",
+    "196": "NON_PARTICIPATING",
     "197": "NON_PARTICIPATING",
     "198": "NON_PARTICIPATING",
     "199": "NON_PARTICIPATING",
     "2": "NON_PARTICIPATING",
     "20": "NON_PARTICIPATING",
-    "200": "FAIL",
+    "200": "NON_PARTICIPATING",
     "201": "NON_PARTICIPATING",
     "202": "NON_PARTICIPATING",
-    "203": "SUCCESS",
+    "203": "NON_PARTICIPATING",
     "204": "NON_PARTICIPATING",
-    "205": "FAIL",
+    "205": "NON_PARTICIPATING",
     "206": "NON_PARTICIPATING",
     "207": "NON_PARTICIPATING",
     "208": "NON_PARTICIPATING",
-    "209": "FAIL",
+    "209": "NON_PARTICIPATING",
     "21": "NON_PARTICIPATING",
     "210": "NON_PARTICIPATING",
-    "211": "SUCCESS",
+    "211": "NON_PARTICIPATING",
     "212": "NON_PARTICIPATING",
-    "213": "SUCCESS",
+    "213": "NON_PARTICIPATING",
     "214": "NON_PARTICIPATING",
     "215": "NON_PARTICIPATING",
     "216": "NON_PARTICIPATING",
     "217": "NON_PARTICIPATING",
     "218": "NON_PARTICIPATING",
     "219": "NON_PARTICIPATING",
-    "22": "FAIL",
+    "22": "NON_PARTICIPATING",
     "220": "NON_PARTICIPATING",
     "221": "NON_PARTICIPATING",
     "222": "NON_PARTICIPATING",
     "223": "NON_PARTICIPATING",
     "224": "NON_PARTICIPATING",
     "225": "NON_PARTICIPATING",
-    "226": "SUCCESS",
+    "226": "NON_PARTICIPATING",
     "227": "NON_PARTICIPATING",
     "228": "NON_PARTICIPATING",
     "229": "NON_PARTICIPATING",
     "23": "NON_PARTICIPATING",
-    "230": "FAIL",
+    "230": "NON_PARTICIPATING",
     "231": "NON_PARTICIPATING",
     "232": "NON_PARTICIPATING",
-    "233": "SUCCESS",
+    "233": "NON_PARTICIPATING",
     "234": "NON_PARTICIPATING",
     "235": "SUCCESS",
     "236": "NON_PARTICIPATING",
-    "237": "FAIL",
-    "238": "FAIL",
-    "239": "FAIL",
+    "237": "NON_PARTICIPATING",
+    "238": "NON_PARTICIPATING",
+    "239": "NON_PARTICIPATING",
     "24": "NON_PARTICIPATING",
     "240": "SUCCESS",
     "241": "NON_PARTICIPATING",
     "242": "NON_PARTICIPATING",
-    "243": "FAIL",
-    "244": "SUCCESS",
+    "243": "NON_PARTICIPATING",
+    "244": "NON_PARTICIPATING",
     "245": "NON_PARTICIPATING",
     "246": "NON_PARTICIPATING",
     "247": "NON_PARTICIPATING",
     "248": "NON_PARTICIPATING",
-    "249": "SUCCESS",
-    "25": "SUCCESS",
+    "249": "NON_PARTICIPATING",
+    "25": "NON_PARTICIPATING",
     "250": "NON_PARTICIPATING",
-    "251": "SUCCESS",
-    "252": "FAIL",
+    "251": "NON_PARTICIPATING",
+    "252": "NON_PARTICIPATING",
     "253": "NON_PARTICIPATING",
     "254": "NON_PARTICIPATING",
     "255": "NON_PARTICIPATING",
@@ -193,36 +193,36 @@
     "38": "NON_PARTICIPATING",
     "39": "NON_PARTICIPATING",
     "4": "NON_PARTICIPATING",
-    "40": "SUCCESS",
+    "40": "NON_PARTICIPATING",
     "41": "NON_PARTICIPATING",
     "42": "NON_PARTICIPATING",
     "43": "NON_PARTICIPATING",
     "44": "NON_PARTICIPATING",
     "45": "NON_PARTICIPATING",
     "46": "NON_PARTICIPATING",
-    "47": "FAIL",
+    "47": "NON_PARTICIPATING",
     "48": "NON_PARTICIPATING",
     "49": "NON_PARTICIPATING",
     "5": "NON_PARTICIPATING",
     "50": "NON_PARTICIPATING",
-    "51": "SUCCESS",
+    "51": "NON_PARTICIPATING",
     "52": "NON_PARTICIPATING",
     "53": "NON_PARTICIPATING",
     "54": "NON_PARTICIPATING",
     "55": "NON_PARTICIPATING",
-    "56": "SUCCESS",
+    "56": "NON_PARTICIPATING",
     "57": "NON_PARTICIPATING",
     "58": "SUCCESS",
     "59": "NON_PARTICIPATING",
-    "6": "FAIL",
+    "6": "NON_PARTICIPATING",
     "60": "NON_PARTICIPATING",
     "61": "NON_PARTICIPATING",
-    "62": "SUCCESS",
+    "62": "NON_PARTICIPATING",
     "63": "NON_PARTICIPATING",
     "64": "NON_PARTICIPATING",
     "65": "NON_PARTICIPATING",
     "66": "NON_PARTICIPATING",
-    "67": "SUCCESS",
+    "67": "NON_PARTICIPATING",
     "68": "NON_PARTICIPATING",
     "69": "NON_PARTICIPATING",
     "7": "NON_PARTICIPATING",
@@ -230,8 +230,8 @@
     "71": "NON_PARTICIPATING",
     "72": "NON_PARTICIPATING",
     "73": "NON_PARTICIPATING",
-    "74": "FAIL",
-    "75": "FAIL",
+    "74": "NON_PARTICIPATING",
+    "75": "NON_PARTICIPATING",
     "76": "NON_PARTICIPATING",
     "77": "NON_PARTICIPATING",
     "78": "NON_PARTICIPATING",
@@ -240,24 +240,24 @@
     "80": "NON_PARTICIPATING",
     "81": "NON_PARTICIPATING",
     "82": "NON_PARTICIPATING",
-    "83": "SUCCESS",
-    "84": "FAIL",
+    "83": "NON_PARTICIPATING",
+    "84": "NON_PARTICIPATING",
     "85": "NON_PARTICIPATING",
-    "86": "SUCCESS",
+    "86": "NON_PARTICIPATING",
     "87": "NON_PARTICIPATING",
     "88": "NON_PARTICIPATING",
     "89": "NON_PARTICIPATING",
     "9": "NON_PARTICIPATING",
-    "90": "SUCCESS",
+    "90": "NON_PARTICIPATING",
     "91": "NON_PARTICIPATING",
     "92": "NON_PARTICIPATING",
     "93": "NON_PARTICIPATING",
-    "94": "FAIL",
-    "95": "SUCCESS",
+    "94": "NON_PARTICIPATING",
+    "95": "NON_PARTICIPATING",
     "96": "NON_PARTICIPATING",
-    "97": "SUCCESS",
+    "97": "NON_PARTICIPATING",
     "98": "NON_PARTICIPATING",
-    "99": "SUCCESS"
+    "99": "NON_PARTICIPATING"
   },
   "architectures": [
     "GPTOptim"
@@ -268,25 +268,25 @@
     "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
   },
   "block_list": [
-    5148709,
-    5148714,
-    5148719,
-    5148724,
-    5148728,
-    5148733,
-    5148738,
-    5148742,
-    5148747,
-    5148751
+    5148841,
+    5148846,
+    5148850,
+    5148855,
+    5148860,
+    5148864,
+    5148869,
+    5148874,
+    5148878,
+    5148883
   ],
   "block_size": 1024,
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,
-  "inner_step": 238,
+  "inner_step": 5,
   "inner_steps": 0,
-  "last_allreduce_block": 5146835,
+  "last_allreduce_block": 5148769,
   "layer_norm_epsilon": 1e-05,
   "model_type": "gpt_optimized",
   "n_embd": 1280,

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:59c255261461b163c0bc0d15396a62a2e2204c16148e5b6076a571ce12efd1a7
+oid sha256:71e4f0a3c3d015ccbe172d2987a3b997b207ec0448344c8b55f409b6b561fd3a
 size 8081781770

model.safetensors

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