Update luigi checkpoint + metadata

luigi/cat_maps.json

@@ -3,9 +3,7 @@
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj7_subtype": {
     "-1": 0,
@@ -13,15 +11,15 @@
     "1": 2,
     "2": 3,
     "3": 4,
-    "4": 5
+    "4": 5,
+    "6": 6,
+    "7": 7
   },
   "proj6_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj6_subtype": {
     "-1": 0,
@@ -30,15 +28,16 @@
     "2": 3,
     "3": 4,
     "4": 5,
-    "7": 6
+    "5": 6,
+    "6": 7,
+    "7": 8,
+    "8": 9
   },
   "proj5_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj5_subtype": {
     "-1": 0,
@@ -56,9 +55,7 @@
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj4_subtype": {
     "-1": 0,
@@ -70,15 +67,14 @@
     "5": 6,
     "6": 7,
     "7": 8,
-    "8": 9
+    "8": 9,
+    "9": 10
   },
   "proj3_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj3_subtype": {
     "-1": 0,
@@ -90,15 +86,14 @@
     "5": 6,
     "6": 7,
     "7": 8,
-    "8": 9
+    "8": 9,
+    "9": 10
   },
   "proj2_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj2_subtype": {
     "-1": 0,
@@ -110,15 +105,14 @@
     "5": 6,
     "6": 7,
     "7": 8,
-    "8": 9
+    "8": 9,
+    "9": 10
   },
   "proj1_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj1_subtype": {
     "-1": 0,
@@ -130,15 +124,14 @@
     "5": 6,
     "6": 7,
     "7": 8,
-    "8": 9
+    "8": 9,
+    "9": 10
   },
   "proj0_owner": {
     "-1": 0,
     "0": 1,
     "1": 2,
-    "2": 3,
-    "3": 4,
-    "4": 5
+    "2": 3
   },
   "proj0_subtype": {
     "-1": 0,
@@ -151,14 +144,14 @@
     "6": 7,
     "7": 8,
     "8": 9,
-    "11": 10
+    "9": 10,
+    "11": 11,
+    "17": 12
   },
   "opp_port": {
     "0": 0,
     "1": 1,
-    "2": 2,
-    "3": 3,
-    "4": 4
+    "2": 2
   },
   "opp_costume": {
     "0": 0,
@@ -166,14 +159,13 @@
     "2": 2,
     "3": 3,
     "4": 4,
-    "5": 5
+    "5": 5,
+    "8": 6
   },
   "self_port": {
     "0": 0,
     "1": 1,
-    "2": 2,
-    "3": 3,
-    "4": 4
+    "2": 2
   },
   "self_costume": {
     "0": 0,
@@ -181,6 +173,7 @@
     "2": 2,
     "3": 3,
     "4": 4,
-    "5": 5
+    "5": 5,
+    "8": 6
   }
 }

luigi/config.json

@@ -4,12 +4,12 @@
   "num_layers": 6,
   "dim_feedforward": 2048,
   "dropout": 0.2,
-  "encoder_type": "hal_flat",
+  "encoder_type": "mimic_flat",
   "d_intra": 256,
   "encoder_nlayers": 2,
   "k_query": 1,
   "scaled_emb": false,
-  "max_seq_len": 256,
+  "max_seq_len": 180,
   "pos_enc": "relpos",
   "rope_theta": 10000.0,
   "rope_learnable_freqs": false,
@@ -18,6 +18,7 @@
   "n_kv_heads": 0,
   "use_rmsnorm": false,
   "use_swiglu": false,
+  "use_input_gate": false,
   "stick_loss": "clusters",
   "btn_loss": "focal",
   "no_opp_inputs": true,
@@ -26,11 +27,14 @@
   "n_stick_clusters": 37,
   "n_shoulder_bins": 3,
   "autoregressive_heads": true,
-  "hal_mode": true,
+  "mimic_mode": true,
   "lean_features": false,
-  "hal_minimal_features": true,
-  "hal_controller_encoding": true,
+  "mimic_minimal_features": false,
+  "mimic_controller_encoding": true,
   "n_controller_combos": 7,
+  "hal_mode": true,
+  "hal_minimal_features": false,
+  "hal_controller_encoding": true,
   "num_stages": 6,
   "num_ports": 4,
   "num_characters": 27,
@@ -39,6 +43,6 @@
   "num_c_dirs": 9,
   "num_proj_types": 103,
   "num_proj_subtypes": 40,
-  "model_preset": "hal",
-  "run_name": "luigi-7class-v2-long"
+  "model_preset": "mimic",
+  "run_name": "luigi-20260420-baseline"
 }

luigi/controller_combos.json

@@ -1,5 +1,5 @@
 {
-  "button_names": ["A", "B", "Z", "JUMP", "TRIG", "A_TRIG", "NONE"],
-  "n_combos": 7,
-  "class_scheme": "melee_7class"
+    "button_names": ["A", "B", "Z", "JUMP", "TRIG", "A_TRIG", "NONE"],
+    "n_combos": 7,
+    "class_scheme": "melee_7class"
 }

luigi/metadata.json

@@ -1,16 +1,16 @@
 {
   "character": "Luigi",
   "melee_enum": "LUIGI",
-  "run_name": "luigi-7class-v2-long",
-  "global_step": 5242,
-  "games_trained": 1951,
-  "val_btn_f1": "~91%",
-  "val_main_f1": "~60%",
-  "val_loss": "~1.0",
-  "n_params": 26276389,
+  "run_name": "luigi-20260420-baseline",
+  "global_step": 5559,
+  "n_params": 26285605,
   "n_controller_combos": 7,
-  "model_preset": "hal",
+  "model_preset": "mimic",
   "pos_enc": "relpos",
-  "no_self_inputs": false,
-  "training_notes": "Trained 2026-04-12 on only 1951 Luigi games (all that was available on the HuggingFace dataset). The training run was set to 262K steps but val loss started climbing around step 5K due to overfitting on the small dataset \u2014 the auto-saved _best.pt is from step 5242, the early-stop sweet spot. Despite the small dataset, this checkpoint wavedashes at 70-83% conversion rate in bot-vs-bot dittos, matching real high-level Luigi play."
+  "val_loss": "?",
+  "best_step": null,
+  "val_btn_f1": "89.4%",
+  "val_main_f1": "57.3%",
+  "val_shldr_f1": "88.0%",
+  "val_cdir_f1": "61.0%"
 }

luigi/mimic_norm.json

@@ -3,16 +3,16 @@
     "percent": {
       "transform": "normalize",
       "min": 0.0,
-      "max": 248.74000549316406,
-      "mean": 51.95983317255054,
-      "std": 41.968951332585824
+      "max": 253.85003662109375,
+      "mean": 61.64429932689563,
+      "std": 46.34995915235096
     },
     "stock": {
       "transform": "normalize",
       "min": 0.0,
       "max": 4.0,
-      "mean": 2.706313936666291,
-      "std": 1.0676758056121092
+      "mean": 2.664070123753332,
+      "std": 1.0649864403960638
     },
     "facing": {
       "transform": "normalize",
@@ -32,8 +32,8 @@
       "transform": "normalize",
       "min": 0.0,
       "max": 6.0,
-      "mean": 1.37347476621212,
-      "std": 1.0
+      "mean": 1.3340354002681392,
+      "std": 1.0360511297936497
     },
     "on_ground": {
       "transform": "normalize",
@@ -44,24 +44,52 @@
     },
     "shield_strength": {
       "transform": "invert_normalize",
-      "min": 0.0,
+      "min": 0.07000000029802322,
       "max": 60.0,
-      "mean": 58.83739485300921,
-      "std": 3.5948948328876775
+      "mean": 57.904816563473794,
+      "std": 5.321308722427969
     },
     "pos_x": {
       "transform": "standardize",
-      "min": -273.2964782714844,
-      "max": 272.9228515625,
-      "mean": 2.095804605074829,
-      "std": 56.876036164389376
+      "min": -273.8414611816406,
+      "max": 271.49237060546875,
+      "mean": 1.8971677825980526,
+      "std": 55.86802327143212
     },
     "pos_y": {
       "transform": "standardize",
-      "min": -153.08863830566406,
-      "max": 312.3891296386719,
-      "mean": 11.285403637158234,
-      "std": 32.36464426855818
+      "min": -154.46617126464844,
+      "max": 345.9438171386719,
+      "mean": 17.525734886350396,
+      "std": 34.056683622248116
+    },
+    "speed_air_x_self": {
+      "transform": "tanh_scale",
+      "scale": 5.0
+    },
+    "speed_ground_x_self": {
+      "transform": "tanh_scale",
+      "scale": 5.0
+    },
+    "speed_y_self": {
+      "transform": "tanh_scale",
+      "scale": 5.0
+    },
+    "speed_x_attack": {
+      "transform": "tanh_scale",
+      "scale": 10.0
+    },
+    "speed_y_attack": {
+      "transform": "tanh_scale",
+      "scale": 10.0
+    },
+    "hitlag_left": {
+      "transform": "linear_max",
+      "max": 20.0
+    },
+    "hitstun_left": {
+      "transform": "log_max",
+      "max": 120.0
     }
   }
 }

luigi/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bb00597a5ec0c270452b1edff9b34399a51c9061e2892c5bd4a50f35e548243e
-size 265140187
+oid sha256:3be7d6034095e33cac7e693fc648662a1ff058969e95854fb517e2ccc331676d
+size 265254050

luigi/norm_stats.json

@@ -1,43 +1,43 @@
 {
   "blastzone_bottom": [
-    -117.65425407727993,
-    18.262450305928876
+    -116.43577896093538,
+    18.086782242473635
   ],
   "blastzone_left": [
-    -221.0636767362987,
-    26.446253126378878
+    -217.53769665521608,
+    23.955724236862498
   ],
   "blastzone_right": [
-    220.69921511796144,
-    27.07588367452761
+    217.2034447510124,
+    24.544563586983777
   ],
   "blastzone_top": [
-    197.0612914988483,
-    25.107361977908766
+    196.8228081177923,
+    21.381798523163674
   ],
   "distance": [
-    46.89352840468522,
-    41.6457437018197
+    50.30362755279843,
+    43.71903710098504
   ],
   "frame": [
-    4974.586012834133,
-    3231.6958953148887
+    6034.551042353503,
+    4033.393943893162
   ],
   "left_platform_height": [
-    26.38494154164375,
-    2.3727475770040893
+    26.432678118398687,
+    2.0716530597064837
   ],
   "left_platform_left": [
-    -58.22050999659657,
-    2.2432174911865275
+    -58.07564371994205,
+    1.8685204894097867
   ],
   "left_platform_right": [
-    -25.508112173086992,
-    4.376336119755992
+    -24.019661709615605,
+    4.387072050681266
   ],
   "opp_action_frame": [
-    11.87692085782767,
-    12.836349237420098
+    12.454146890384358,
+    13.191764021553729
   ],
   "opp_ecb_bottom_x": [
     0.0,
@@ -64,44 +64,44 @@
     1.0
   ],
   "opp_ecb_top_x": [
-    0.0,
+    1.3155713337021068e-43,
     1.0
   ],
   "opp_ecb_top_y": [
-    0.0,
+    1.1681394676340078e-31,
     1.0
   ],
   "opp_hitlag_left": [
-    0.011022251381279467,
+    0.18935975073332442,
     1.0
   ],
   "opp_hitstun_left": [
-    6.207891702008504,
-    21.609078401117316
+    5.8551569869454285,
+    18.68708366887291
   ],
   "opp_invuln_left": [
     0.0,
     1.0
   ],
   "opp_jumps_left": [
-    1.37347476621212,
-    1.0
+    1.3340354002681392,
+    1.0360511297936497
   ],
   "opp_l_shldr": [
-    0.16628418862819672,
+    0.15142609179019928,
     1.0
   ],
   "opp_main_x": [
-    0.18162456154823303,
+    0.15142609179019928,
     1.0
   ],
   "opp_main_y": [
-    0.18162456154823303,
+    0.15142609179019928,
     1.0
   ],
   "opp_nana_action_frame": [
-    12.621206460649352,
-    14.60026923736515
+    12.866963269291078,
+    14.591636713702371
   ],
   "opp_nana_ecb_bottom_x": [
     0.0,
@@ -128,27 +128,27 @@
     1.0
   ],
   "opp_nana_ecb_top_x": [
-    0.0,
+    1.3290594085393704e-43,
     1.0
   ],
   "opp_nana_ecb_top_y": [
-    0.0,
+    1.7321037283724877e-31,
     1.0
   ],
   "opp_nana_hitlag_left": [
-    0.0056202595617094175,
+    0.12184843727372381,
     1.0
   ],
   "opp_nana_hitstun_left": [
-    5.165146239507015,
-    22.602253744569836
+    8.561676114921209,
+    32.44870503531152
   ],
   "opp_nana_invuln_left": [
     0.0,
     1.0
   ],
   "opp_nana_jumps_left": [
-    1.4042130889419322,
+    1.3998688017707925,
     1.0
   ],
   "opp_nana_l_shldr": [
@@ -164,92 +164,92 @@
     1.0
   ],
   "opp_nana_percent": [
-    39.102286283530674,
-    35.88860614948017
+    45.897926360591185,
+    38.403286549883745
   ],
   "opp_nana_pos_x": [
-    1.2068642679018815,
-    49.30211774003539
+    1.5234565584246424,
+    53.40700962036593
   ],
   "opp_nana_pos_y": [
-    10.58418713398032,
-    34.35033591865463
+    13.790174163373038,
+    34.68172084091242
   ],
   "opp_nana_r_shldr": [
     0.0,
     1.0
   ],
   "opp_nana_shield_strength": [
-    58.85621297161418,
-    3.5665092548631234
+    58.580457119178654,
+    4.296090141085916
   ],
   "opp_nana_speed_air_x_self": [
-    7.449704370577794e-05,
+    0.007791027622784138,
     1.0
   ],
   "opp_nana_speed_ground_x_self": [
-    0.0004617140358648432,
+    0.0036689508973596447,
     1.0
   ],
   "opp_nana_speed_x_attack": [
-    0.0006254869455141354,
+    -0.006111881072818925,
     1.0
   ],
   "opp_nana_speed_y_attack": [
-    0.01099576749089697,
+    0.22299334139315485,
     1.0
   ],
   "opp_nana_speed_y_self": [
-    -0.01155770004031499,
+    -0.27515532009875937,
     1.0
   ],
   "opp_nana_stock": [
-    2.653193168742282,
-    1.0754299773926206
+    2.643058338136582,
+    1.0861104121950493
   ],
   "opp_percent": [
-    51.95983317255058,
-    41.968951332585775
+    61.64429932689556,
+    46.349959152351055
   ],
   "opp_pos_x": [
-    2.095804605074832,
-    56.87603616438966
+    1.8971677825980504,
+    55.86802327143193
   ],
   "opp_pos_y": [
-    11.285403637157915,
-    32.36464426855837
+    17.525734886349444,
+    34.056683622248855
   ],
   "opp_r_shldr": [
-    0.16628418862819672,
+    0.15142609179019928,
     1.0
   ],
   "opp_shield_strength": [
-    58.83739485300921,
-    3.594894832887488
+    57.904816563473794,
+    5.321308722428183
   ],
   "opp_speed_air_x_self": [
-    -2.5396470322689237e-05,
+    0.0015555543101372434,
     1.0
   ],
   "opp_speed_ground_x_self": [
-    7.379903611212253e-05,
+    0.0008303092686374707,
     1.0
   ],
   "opp_speed_x_attack": [
-    0.00016006144034171966,
+    0.0005406117000125548,
     1.0
   ],
   "opp_speed_y_attack": [
-    0.016839584911564988,
+    0.25476843954116923,
     1.0
   ],
   "opp_speed_y_self": [
-    -0.02085563252216532,
-    1.0
+    -0.29197346092832793,
+    1.2091348456076993
   ],
   "opp_stock": [
-    2.706313936666291,
-    1.0676758056121092
+    2.664070123753332,
+    1.0649864403960638
   ],
   "proj0_frame": [
     0.0,
@@ -412,32 +412,32 @@
     1.0
   ],
   "randall_height": [
-    -23.612431513793055,
-    9.495261087482037
+    -23.598991837765116,
+    9.494447796275164
   ],
   "randall_left": [
-    -4.50880393479985,
-    60.44143162238004
+    -4.714263831591581,
+    60.51851336884743
   ],
   "randall_right": [
-    7.391196065174139,
-    60.441431622283126
+    7.1857361683878125,
+    60.51851336866065
   ],
   "right_platform_height": [
-    26.405558774032343,
-    2.4055154064281274
+    26.445690568737735,
+    2.0950950845682446
   ],
   "right_platform_left": [
-    25.503635359959812,
-    4.369980335853813
+    24.01683619417315,
+    4.382112442340251
   ],
   "right_platform_right": [
-    58.56582543252829,
-    2.7720777242750203
+    58.29358761642036,
+    2.292665540348235
   ],
   "self_action_frame": [
-    11.87692085782767,
-    12.836349237420098
+    12.454146890384358,
+    13.191764021553729
   ],
   "self_ecb_bottom_x": [
     0.0,
@@ -464,44 +464,44 @@
     1.0
   ],
   "self_ecb_top_x": [
-    0.0,
+    1.3155713337021068e-43,
     1.0
   ],
   "self_ecb_top_y": [
-    0.0,
+    1.168139467634003e-31,
     1.0
   ],
   "self_hitlag_left": [
-    0.011022251381279467,
+    0.18935975073332442,
     1.0
   ],
   "self_hitstun_left": [
-    6.207891702008504,
-    21.609078401117316
+    5.8551569869454285,
+    18.68708366887291
   ],
   "self_invuln_left": [
     0.0,
     1.0
   ],
   "self_jumps_left": [
-    1.37347476621212,
-    1.0
+    1.3340354002681392,
+    1.0360511297936497
   ],
   "self_l_shldr": [
-    0.16628387570381165,
+    0.15142609179019928,
     1.0
   ],
   "self_main_x": [
-    0.18162456154823303,
+    0.15142609179019928,
     1.0
   ],
   "self_main_y": [
-    0.18162456154823303,
+    0.15142609179019928,
     1.0
   ],
   "self_nana_action_frame": [
-    12.621206460649352,
-    14.60026923736515
+    12.866963269291078,
+    14.591636713702371
   ],
   "self_nana_ecb_bottom_x": [
     0.0,
@@ -528,27 +528,27 @@
     1.0
   ],
   "self_nana_ecb_top_x": [
-    0.0,
+    1.3290594085393704e-43,
     1.0
   ],
   "self_nana_ecb_top_y": [
-    0.0,
+    1.7321037283724877e-31,
     1.0
   ],
   "self_nana_hitlag_left": [
-    0.0056202595617094175,
+    0.12184843727372381,
     1.0
   ],
   "self_nana_hitstun_left": [
-    5.165146239507015,
-    22.602253744569836
+    8.561676114921209,
+    32.44870503531152
   ],
   "self_nana_invuln_left": [
     0.0,
     1.0
   ],
   "self_nana_jumps_left": [
-    1.4042130889419322,
+    1.3998688017707925,
     1.0
   ],
   "self_nana_l_shldr": [
@@ -564,111 +564,111 @@
     1.0
   ],
   "self_nana_percent": [
-    39.102286283530674,
-    35.88860614948017
+    45.897926360591185,
+    38.403286549883745
   ],
   "self_nana_pos_x": [
-    1.2068642679018815,
-    49.30211774003539
+    1.5234565584246424,
+    53.40700962036593
   ],
   "self_nana_pos_y": [
-    10.58418713398032,
-    34.35033591865463
+    13.790174163373038,
+    34.68172084091242
   ],
   "self_nana_r_shldr": [
     0.0,
     1.0
   ],
   "self_nana_shield_strength": [
-    58.85621297161418,
-    3.5665092548631234
+    58.580457119178654,
+    4.296090141085916
   ],
   "self_nana_speed_air_x_self": [
-    7.449704370577794e-05,
+    0.007791027622784138,
     1.0
   ],
   "self_nana_speed_ground_x_self": [
-    0.0004617140358648432,
+    0.0036689508973596447,
     1.0
   ],
   "self_nana_speed_x_attack": [
-    0.0006254869455141354,
+    -0.006111881072818925,
     1.0
   ],
   "self_nana_speed_y_attack": [
-    0.01099576749089697,
+    0.22299334139315485,
     1.0
   ],
   "self_nana_speed_y_self": [
-    -0.01155770004031499,
+    -0.27515532009875937,
     1.0
   ],
   "self_nana_stock": [
-    2.653193168742282,
-    1.0754299773926206
+    2.643058338136582,
+    1.0861104121950493
   ],
   "self_percent": [
-    51.95983317255054,
-    41.968951332585824
+    61.64429932689563,
+    46.34995915235096
   ],
   "self_pos_x": [
-    2.095804605074829,
-    56.876036164389376
+    1.8971677825980526,
+    55.86802327143212
   ],
   "self_pos_y": [
-    11.285403637158234,
-    32.36464426855818
+    17.525734886350396,
+    34.056683622248116
   ],
   "self_r_shldr": [
-    0.16628387570381165,
+    0.15142609179019928,
     1.0
   ],
   "self_shield_strength": [
-    58.83739485300921,
-    3.5948948328876775
+    57.904816563473794,
+    5.321308722427969
   ],
   "self_speed_air_x_self": [
-    -2.5396470322689227e-05,
+    0.0015555543101372434,
     1.0
   ],
   "self_speed_ground_x_self": [
-    7.379903611212253e-05,
+    0.0008303092686374707,
     1.0
   ],
   "self_speed_x_attack": [
-    0.00016006144034171966,
+    0.0005406117000125548,
     1.0
   ],
   "self_speed_y_attack": [
-    0.016839584911564988,
+    0.2547684395411692,
     1.0
   ],
   "self_speed_y_self": [
-    -0.02085563252216532,
-    1.0
+    -0.29197346092832777,
+    1.2091348456077005
   ],
   "self_stock": [
-    2.706313936666291,
-    1.0676758056121092
+    2.664070123753332,
+    1.0649864403960638
   ],
   "stage_edge_left": [
-    -75.58011646892624,
-    11.397204097224181
+    -73.05073560293425,
+    10.107829832368209
   ],
   "stage_edge_right": [
-    75.58011646892624,
-    11.397204097224181
+    73.05073560293425,
+    10.107829832368209
   ],
   "top_platform_height": [
-    48.35880538463942,
-    5.535745808270807
+    49.10384879310862,
+    5.688576976238595
   ],
   "top_platform_left": [
-    -17.193803214501965,
-    1.9480815914430683
+    -17.256542911891106,
+    1.9637217314969677
   ],
   "top_platform_right": [
-    17.19358611770908,
-    1.947878301464183
+    17.25642096071091,
+    1.9636123586957406
   ]
 }