metadata
license: other
tags:
- robotics
- fastwam
- flashwam
- place-cube
pcbnew_flashwam_ft — FusedKV / RoPE-fixed FlashWAM, finetuned on place_cube_new (500 traj)
FlashWAM (M1_FusedKV_RopeFixed) action-video model, finetuned on the new
500-episode place-cube-in-bowl dataset ("pick and place new").
- Model: FasterWAM decoupled,
kv_source_mode: fused_kv,fixed_rope: true(video DiT Wan2.2-TI2V-5B backbone + 1-layer action DiT). - Data:
place_cube_new_lerobot_v21— 500 episodes / 51,579 frames, 10 Hz, 2 cameras (base + wrist), 7-dim delta action, 8-dim proprio (real gripper widths). Task string: "place the cube in the bowl". - Init: finetuned from LIBERO checkpoint
M1_FusedKV_RopeFixed_step021700(new fused_kv format). - Training: 30 epochs / 48,360 steps, 4×H200, global batch 32, lr 1e-4 cosine,
bf16. Final
loss=0.0802,loss_action=0.0045.
Checkpoints (checkpoints/weights/)
Weights-only checkpoints saved every 5 epochs:
| file | step | epoch |
|---|---|---|
step_008060.pt |
8,060 | 5 |
step_016120.pt |
16,120 | 10 |
step_024180.pt |
24,180 | 15 |
step_032240.pt |
32,240 | 20 |
step_040300.pt |
40,300 | 25 |
step_048360.pt |
48,360 | 30 (final) |
Each file is ~10.13 GB (bf16 full model: fused video + action experts).
Other files
config.yaml— full training/model config for this run.dataset_stats.json— per-run action/state normalization stats (min/max), needed at inference for de/normalization.