Pepijn b81eef43c8 fix(profiling): wall_x OOM and xvla rename_map ...

- wall_x: switch to SGD optimizer + explicit scheduler overrides.
  The 4B-param model casts to bf16 internally, but AdamW's exp_avg/
  exp_avg_sq states blow past the 22 GB GPU. Same fix we applied to
  pi0/pi05/pi0_fast.
- xvla: fix rename_map. Dataset (libero_plus) exposes front/wrist
  image keys; the model expects image/image2. Previous map was
  direction-reversed and left the batch without any recognized
  image feature.

Made-with: Cursor

2026-04-16 19:49:12 +02:00

..

model_profiling_specs.json

fix(profiling): wall_x OOM and xvla rename_map

2026-04-16 19:49:12 +02:00