Run 5. Outer Step 0. Inner Step 2.

config.json

@@ -1,302 +1,293 @@
 {
-  "_name_or_path": "distributed/optimized-gpt2-1b",
-  "activation_function": "gelu_new",
+  "_name_or_path": "distributed/llama-1b",
   "all_reduce_scores": {
-    "0": "NON_PARTICIPATING",
+    "0": "SUCCESS",
     "1": "SUCCESS",
-    "10": "NON_PARTICIPATING",
-    "100": "NON_PARTICIPATING",
-    "101": "NON_PARTICIPATING",
-    "102": "NON_PARTICIPATING",
-    "103": "NON_PARTICIPATING",
+    "10": "SUCCESS",
+    "100": "SUCCESS",
+    "101": "SUCCESS",
+    "102": "SUCCESS",
+    "103": "SUCCESS",
     "104": "SUCCESS",
-    "105": "NON_PARTICIPATING",
-    "106": "NON_PARTICIPATING",
-    "107": "NON_PARTICIPATING",
-    "108": "NON_PARTICIPATING",
-    "109": "NON_PARTICIPATING",
+    "105": "SUCCESS",
+    "106": "SUCCESS",
+    "107": "SUCCESS",
+    "108": "SUCCESS",
+    "109": "SUCCESS",
     "11": "SUCCESS",
-    "110": "NON_PARTICIPATING",
-    "111": "NON_PARTICIPATING",
-    "112": "NON_PARTICIPATING",
-    "113": "NON_PARTICIPATING",
-    "114": "NON_PARTICIPATING",
+    "110": "SUCCESS",
+    "111": "SUCCESS",
+    "112": "SUCCESS",
+    "113": "SUCCESS",
+    "114": "SUCCESS",
     "115": "SUCCESS",
-    "116": "NON_PARTICIPATING",
-    "117": "NON_PARTICIPATING",
-    "118": "NON_PARTICIPATING",
-    "119": "NON_PARTICIPATING",
+    "116": "SUCCESS",
+    "117": "SUCCESS",
+    "118": "SUCCESS",
+    "119": "SUCCESS",
     "12": "SUCCESS",
-    "120": "NON_PARTICIPATING",
-    "121": "NON_PARTICIPATING",
-    "122": "NON_PARTICIPATING",
-    "123": "NON_PARTICIPATING",
-    "124": "NON_PARTICIPATING",
-    "125": "NON_PARTICIPATING",
-    "126": "NON_PARTICIPATING",
-    "127": "NON_PARTICIPATING",
-    "128": "NON_PARTICIPATING",
-    "129": "NON_PARTICIPATING",
+    "120": "SUCCESS",
+    "121": "SUCCESS",
+    "122": "SUCCESS",
+    "123": "SUCCESS",
+    "124": "SUCCESS",
+    "125": "SUCCESS",
+    "126": "SUCCESS",
+    "127": "SUCCESS",
+    "128": "SUCCESS",
+    "129": "SUCCESS",
     "13": "SUCCESS",
-    "130": "NON_PARTICIPATING",
-    "131": "NON_PARTICIPATING",
-    "132": "NON_PARTICIPATING",
-    "133": "NON_PARTICIPATING",
-    "134": "NON_PARTICIPATING",
-    "135": "NON_PARTICIPATING",
-    "136": "NON_PARTICIPATING",
-    "137": "NON_PARTICIPATING",
-    "138": "NON_PARTICIPATING",
-    "139": "NON_PARTICIPATING",
+    "130": "SUCCESS",
+    "131": "SUCCESS",
+    "132": "SUCCESS",
+    "133": "SUCCESS",
+    "134": "SUCCESS",
+    "135": "SUCCESS",
+    "136": "SUCCESS",
+    "137": "SUCCESS",
+    "138": "SUCCESS",
+    "139": "SUCCESS",
     "14": "SUCCESS",
     "140": "SUCCESS",
-    "141": "NON_PARTICIPATING",
-    "142": "FAIL",
+    "141": "SUCCESS",
+    "142": "SUCCESS",
     "143": "SUCCESS",
-    "144": "NON_PARTICIPATING",
-    "145": "NON_PARTICIPATING",
-    "146": "NON_PARTICIPATING",
-    "147": "NON_PARTICIPATING",
+    "144": "SUCCESS",
+    "145": "SUCCESS",
+    "146": "SUCCESS",
+    "147": "SUCCESS",
     "148": "SUCCESS",
-    "149": "NON_PARTICIPATING",
-    "15": "NON_PARTICIPATING",
-    "150": "NON_PARTICIPATING",
-    "151": "NON_PARTICIPATING",
-    "152": "NON_PARTICIPATING",
-    "153": "NON_PARTICIPATING",
-    "154": "NON_PARTICIPATING",
-    "155": "NON_PARTICIPATING",
+    "149": "SUCCESS",
+    "15": "SUCCESS",
+    "150": "SUCCESS",
+    "151": "SUCCESS",
+    "152": "SUCCESS",
+    "153": "SUCCESS",
+    "154": "SUCCESS",
+    "155": "SUCCESS",
     "156": "SUCCESS",
-    "157": "NON_PARTICIPATING",
-    "158": "NON_PARTICIPATING",
-    "159": "NON_PARTICIPATING",
+    "157": "SUCCESS",
+    "158": "SUCCESS",
+    "159": "SUCCESS",
     "16": "SUCCESS",
-    "160": "NON_PARTICIPATING",
-    "161": "NON_PARTICIPATING",
-    "162": "NON_PARTICIPATING",
-    "163": "NON_PARTICIPATING",
-    "164": "NON_PARTICIPATING",
-    "165": "NON_PARTICIPATING",
-    "166": "NON_PARTICIPATING",
-    "167": "NON_PARTICIPATING",
-    "168": "NON_PARTICIPATING",
-    "169": "NON_PARTICIPATING",
+    "160": "SUCCESS",
+    "161": "SUCCESS",
+    "162": "SUCCESS",
+    "163": "SUCCESS",
+    "164": "SUCCESS",
+    "165": "SUCCESS",
+    "166": "SUCCESS",
+    "167": "SUCCESS",
+    "168": "SUCCESS",
+    "169": "SUCCESS",
     "17": "SUCCESS",
-    "170": "NON_PARTICIPATING",
-    "171": "NON_PARTICIPATING",
-    "172": "NON_PARTICIPATING",
-    "173": "NON_PARTICIPATING",
-    "174": "NON_PARTICIPATING",
-    "175": "NON_PARTICIPATING",
-    "176": "NON_PARTICIPATING",
-    "177": "NON_PARTICIPATING",
-    "178": "NON_PARTICIPATING",
+    "170": "SUCCESS",
+    "171": "SUCCESS",
+    "172": "SUCCESS",
+    "173": "SUCCESS",
+    "174": "SUCCESS",
+    "175": "SUCCESS",
+    "176": "SUCCESS",
+    "177": "SUCCESS",
+    "178": "SUCCESS",
     "179": "SUCCESS",
     "18": "SUCCESS",
     "180": "SUCCESS",
-    "181": "NON_PARTICIPATING",
-    "182": "NON_PARTICIPATING",
-    "183": "NON_PARTICIPATING",
-    "184": "NON_PARTICIPATING",
-    "185": "NON_PARTICIPATING",
-    "186": "NON_PARTICIPATING",
-    "187": "NON_PARTICIPATING",
-    "188": "NON_PARTICIPATING",
-    "189": "NON_PARTICIPATING",
-    "19": "NON_PARTICIPATING",
-    "190": "NON_PARTICIPATING",
-    "191": "NON_PARTICIPATING",
-    "192": "NON_PARTICIPATING",
-    "193": "NON_PARTICIPATING",
+    "181": "SUCCESS",
+    "182": "SUCCESS",
+    "183": "SUCCESS",
+    "184": "SUCCESS",
+    "185": "SUCCESS",
+    "186": "SUCCESS",
+    "187": "SUCCESS",
+    "188": "SUCCESS",
+    "189": "SUCCESS",
+    "19": "SUCCESS",
+    "190": "SUCCESS",
+    "191": "SUCCESS",
+    "192": "SUCCESS",
+    "193": "SUCCESS",
     "194": "SUCCESS",
-    "195": "NON_PARTICIPATING",
-    "196": "NON_PARTICIPATING",
-    "197": "NON_PARTICIPATING",
-    "198": "NON_PARTICIPATING",
-    "199": "NON_PARTICIPATING",
+    "195": "SUCCESS",
+    "196": "SUCCESS",
+    "197": "SUCCESS",
+    "198": "SUCCESS",
+    "199": "SUCCESS",
     "2": "SUCCESS",
     "20": "SUCCESS",
-    "200": "NON_PARTICIPATING",
-    "201": "NON_PARTICIPATING",
-    "202": "NON_PARTICIPATING",
+    "200": "SUCCESS",
+    "201": "SUCCESS",
+    "202": "SUCCESS",
     "203": "SUCCESS",
-    "204": "NON_PARTICIPATING",
-    "205": "NON_PARTICIPATING",
-    "206": "NON_PARTICIPATING",
-    "207": "NON_PARTICIPATING",
-    "208": "NON_PARTICIPATING",
-    "209": "NON_PARTICIPATING",
+    "204": "SUCCESS",
+    "205": "SUCCESS",
+    "206": "SUCCESS",
+    "207": "SUCCESS",
+    "208": "SUCCESS",
+    "209": "SUCCESS",
     "21": "SUCCESS",
-    "210": "NON_PARTICIPATING",
-    "211": "NON_PARTICIPATING",
-    "212": "NON_PARTICIPATING",
-    "213": "NON_PARTICIPATING",
-    "214": "NON_PARTICIPATING",
-    "215": "NON_PARTICIPATING",
-    "216": "NON_PARTICIPATING",
-    "217": "NON_PARTICIPATING",
-    "218": "NON_PARTICIPATING",
-    "219": "NON_PARTICIPATING",
+    "210": "SUCCESS",
+    "211": "SUCCESS",
+    "212": "SUCCESS",
+    "213": "SUCCESS",
+    "214": "SUCCESS",
+    "215": "SUCCESS",
+    "216": "SUCCESS",
+    "217": "SUCCESS",
+    "218": "SUCCESS",
+    "219": "SUCCESS",
     "22": "SUCCESS",
-    "220": "NON_PARTICIPATING",
-    "221": "NON_PARTICIPATING",
-    "222": "NON_PARTICIPATING",
-    "223": "NON_PARTICIPATING",
-    "224": "NON_PARTICIPATING",
-    "225": "NON_PARTICIPATING",
-    "226": "NON_PARTICIPATING",
-    "227": "NON_PARTICIPATING",
-    "228": "NON_PARTICIPATING",
-    "229": "NON_PARTICIPATING",
+    "220": "SUCCESS",
+    "221": "SUCCESS",
+    "222": "SUCCESS",
+    "223": "SUCCESS",
+    "224": "SUCCESS",
+    "225": "SUCCESS",
+    "226": "SUCCESS",
+    "227": "SUCCESS",
+    "228": "SUCCESS",
+    "229": "SUCCESS",
     "23": "SUCCESS",
-    "230": "NON_PARTICIPATING",
-    "231": "NON_PARTICIPATING",
-    "232": "NON_PARTICIPATING",
-    "233": "NON_PARTICIPATING",
-    "234": "NON_PARTICIPATING",
-    "235": "NON_PARTICIPATING",
-    "236": "NON_PARTICIPATING",
-    "237": "NON_PARTICIPATING",
-    "238": "NON_PARTICIPATING",
-    "239": "NON_PARTICIPATING",
+    "230": "SUCCESS",
+    "231": "SUCCESS",
+    "232": "SUCCESS",
+    "233": "SUCCESS",
+    "234": "SUCCESS",
+    "235": "SUCCESS",
+    "236": "SUCCESS",
+    "237": "SUCCESS",
+    "238": "SUCCESS",
+    "239": "SUCCESS",
     "24": "SUCCESS",
-    "240": "NON_PARTICIPATING",
-    "241": "NON_PARTICIPATING",
-    "242": "NON_PARTICIPATING",
-    "243": "NON_PARTICIPATING",
-    "244": "NON_PARTICIPATING",
-    "245": "NON_PARTICIPATING",
-    "246": "NON_PARTICIPATING",
-    "247": "NON_PARTICIPATING",
-    "248": "NON_PARTICIPATING",
-    "249": "NON_PARTICIPATING",
+    "240": "SUCCESS",
+    "241": "SUCCESS",
+    "242": "SUCCESS",
+    "243": "SUCCESS",
+    "244": "SUCCESS",
+    "245": "SUCCESS",
+    "246": "SUCCESS",
+    "247": "SUCCESS",
+    "248": "SUCCESS",
+    "249": "SUCCESS",
     "25": "SUCCESS",
-    "250": "NON_PARTICIPATING",
-    "251": "NON_PARTICIPATING",
-    "252": "NON_PARTICIPATING",
-    "253": "NON_PARTICIPATING",
-    "254": "NON_PARTICIPATING",
-    "255": "NON_PARTICIPATING",
-    "26": "NON_PARTICIPATING",
+    "250": "SUCCESS",
+    "251": "SUCCESS",
+    "252": "SUCCESS",
+    "253": "SUCCESS",
+    "254": "SUCCESS",
+    "255": "SUCCESS",
+    "26": "SUCCESS",
     "27": "SUCCESS",
     "28": "SUCCESS",
     "29": "SUCCESS",
     "3": "SUCCESS",
-    "30": "NON_PARTICIPATING",
+    "30": "SUCCESS",
     "31": "SUCCESS",
     "32": "SUCCESS",
-    "33": "NON_PARTICIPATING",
-    "34": "NON_PARTICIPATING",
+    "33": "SUCCESS",
+    "34": "SUCCESS",
     "35": "SUCCESS",
-    "36": "NON_PARTICIPATING",
+    "36": "SUCCESS",
     "37": "SUCCESS",
-    "38": "NON_PARTICIPATING",
+    "38": "SUCCESS",
     "39": "SUCCESS",
-    "4": "NON_PARTICIPATING",
-    "40": "NON_PARTICIPATING",
-    "41": "NON_PARTICIPATING",
-    "42": "NON_PARTICIPATING",
-    "43": "NON_PARTICIPATING",
+    "4": "SUCCESS",
+    "40": "SUCCESS",
+    "41": "SUCCESS",
+    "42": "SUCCESS",
+    "43": "SUCCESS",
     "44": "SUCCESS",
     "45": "SUCCESS",
     "46": "SUCCESS",
-    "47": "NON_PARTICIPATING",
-    "48": "NON_PARTICIPATING",
+    "47": "SUCCESS",
+    "48": "SUCCESS",
     "49": "SUCCESS",
-    "5": "NON_PARTICIPATING",
+    "5": "SUCCESS",
     "50": "SUCCESS",
     "51": "SUCCESS",
-    "52": "NON_PARTICIPATING",
+    "52": "SUCCESS",
     "53": "SUCCESS",
     "54": "SUCCESS",
-    "55": "NON_PARTICIPATING",
-    "56": "NON_PARTICIPATING",
+    "55": "SUCCESS",
+    "56": "SUCCESS",
     "57": "SUCCESS",
-    "58": "NON_PARTICIPATING",
-    "59": "NON_PARTICIPATING",
+    "58": "SUCCESS",
+    "59": "SUCCESS",
     "6": "SUCCESS",
     "60": "SUCCESS",
     "61": "SUCCESS",
     "62": "SUCCESS",
     "63": "SUCCESS",
-    "64": "NON_PARTICIPATING",
-    "65": "NON_PARTICIPATING",
-    "66": "NON_PARTICIPATING",
+    "64": "SUCCESS",
+    "65": "SUCCESS",
+    "66": "SUCCESS",
     "67": "SUCCESS",
     "68": "SUCCESS",
     "69": "SUCCESS",
     "7": "SUCCESS",
-    "70": "NON_PARTICIPATING",
+    "70": "SUCCESS",
     "71": "SUCCESS",
-    "72": "NON_PARTICIPATING",
+    "72": "SUCCESS",
     "73": "SUCCESS",
-    "74": "NON_PARTICIPATING",
-    "75": "NON_PARTICIPATING",
+    "74": "SUCCESS",
+    "75": "SUCCESS",
     "76": "SUCCESS",
     "77": "SUCCESS",
     "78": "SUCCESS",
-    "79": "NON_PARTICIPATING",
-    "8": "NON_PARTICIPATING",
-    "80": "NON_PARTICIPATING",
+    "79": "SUCCESS",
+    "8": "SUCCESS",
+    "80": "SUCCESS",
     "81": "SUCCESS",
-    "82": "NON_PARTICIPATING",
-    "83": "NON_PARTICIPATING",
+    "82": "SUCCESS",
+    "83": "SUCCESS",
     "84": "SUCCESS",
-    "85": "NON_PARTICIPATING",
-    "86": "NON_PARTICIPATING",
+    "85": "SUCCESS",
+    "86": "SUCCESS",
     "87": "SUCCESS",
     "88": "SUCCESS",
     "89": "SUCCESS",
-    "9": "NON_PARTICIPATING",
+    "9": "SUCCESS",
     "90": "SUCCESS",
     "91": "SUCCESS",
-    "92": "NON_PARTICIPATING",
+    "92": "SUCCESS",
     "93": "SUCCESS",
     "94": "SUCCESS",
     "95": "SUCCESS",
-    "96": "NON_PARTICIPATING",
-    "97": "NON_PARTICIPATING",
+    "96": "SUCCESS",
+    "97": "SUCCESS",
     "98": "SUCCESS",
-    "99": "NON_PARTICIPATING"
+    "99": "SUCCESS"
   },
   "architectures": [
-    "GPTOptim"
+    "LlamaForCausalLM"
   ],
-  "attn_pdrop": 0.1,
-  "auto_map": {
-    "AutoConfig": "distributed/optimized-gpt2-500m--configuration_gpt_optimized.GPTOptimConfig",
-    "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
-  },
+  "attention_bias": false,
+  "attention_dropout": 0.0,
   "block_list": [
-    5699638,
-    5699671
+    5701406,
+    5701451
   ],
-  "block_size": 1024,
-  "bos_token_id": 50256,
-  "embd_pdrop": 0.1,
-  "eos_token_id": 50256,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
   "initializer_range": 0.02,
-  "inner_step": 4,
-  "inner_steps": 0,
-  "last_allreduce_block": 5696880,
-  "layer_norm_epsilon": 1e-05,
-  "model_type": "gpt_optimized",
-  "n_embd": 1280,
-  "n_head": 32,
-  "n_inner": null,
-  "n_layer": 48,
-  "n_positions": 1024,
-  "reorder_and_upcast_attn": false,
-  "resid_pdrop": 0.1,
-  "scale_attn_by_inverse_layer_idx": false,
-  "scale_attn_weights": true,
-  "summary_activation": null,
-  "summary_first_dropout": 0.1,
-  "summary_proj_to_labels": true,
-  "summary_type": "cls_index",
-  "summary_use_proj": true,
+  "inner_step": 2,
+  "intermediate_size": 5632,
+  "last_allreduce_block": 5699671,
+  "max_position_embeddings": 2048,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 22,
+  "num_key_value_heads": 4,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
   "torch_dtype": "float32",
   "transformers_version": "4.39.3",
-  "use_cache": true,
-  "vocab_size": 50257
+  "use_cache": false,
+  "vocab_size": 32000
 }

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.39.3",
+  "use_cache": false
+}

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a7c61747f2dea043eb460af9daaa47a54422d0058a1d5423ecc7fb99da965840
-size 8081782026
+oid sha256:47b1d246a43ceccac4efd391b16ab3e2b0722fc1eebc3a0d5dec97f7bee5f66a
+size 8800564561

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:365e1ded998369e71e73e7dadad5007283c330b7354fd7d30a12baedf6034f93
-size 4040701744
+oid sha256:013c488772d9a888bfbe950fd8934c089110819b407450179d5afdf633308df8
+size 4400216536