| { | |
| "model": "MarioStudent", | |
| "architecture": "MobileNetV3-Large (ImageNet init) + heads: move (11), survival (11), tile grid 13x12, enemy grid 13x12", | |
| "parameters": 9035296, | |
| "input": { | |
| "name": "crop", | |
| "shape": [ | |
| 1, | |
| 6, | |
| 120, | |
| 96 | |
| ], | |
| "dtype": "float16", | |
| "range": [ | |
| 0, | |
| 1 | |
| ], | |
| "layout": "RGB now, then RGB 4 frames earlier; screen from 32 px behind Mario to 160 px ahead, every 2nd pixel" | |
| }, | |
| "output": { | |
| "name": "logits", | |
| "meaning": "11 move logits (order in `moves`) then 11 survival logits" | |
| }, | |
| "moves": [ | |
| "run", | |
| "walk", | |
| "hold", | |
| "jump", | |
| "hop", | |
| "up", | |
| "left", | |
| "leftjump", | |
| "wait", | |
| "swim", | |
| "down" | |
| ], | |
| "macro_frames": 8, | |
| "coreml": "MarioStudent.mlpackage (fp16, 185/185 ops on the Neural Engine)", | |
| "pytorch": "mario-student.pt (model-lab mario-tiny student_retro.py, arch retro-aux, view half2)" | |
| } |