Skip to content

SmolVLA training configs and SPA episode viewer#339

Closed
vertix wants to merge 5 commits intoPositronic-Robotics:mainfrom
vertix:worktree-smolvla
Closed

SmolVLA training configs and SPA episode viewer#339
vertix wants to merge 5 commits intoPositronic-Robotics:mainfrom
vertix:worktree-smolvla

Conversation

@vertix
Copy link
Copy Markdown
Contributor

@vertix vertix commented Mar 7, 2026

Summary

  • Fix SmolVLA training: correct config import (SmolVLAConfig instead of XVLAConfig), set recommended batch_size=64, fix LR scheduler decay, add jinja2 dep
  • Add full_finetune training config (unfreezes vision encoder + all params) alongside default expert-only train
  • SPA episode navigation in viewer: uses rerun's add_receiver/remove_receiver API to swap RRD data without reloading the 35MB WASM. Adds Service Worker for WASM caching and /api/episode/{id} JSON endpoint

Test plan

  • Ran both train and full_finetune on sim_stack dataset (60K steps each on H100)
  • Ran sim inference (10 episodes each) against both checkpoints
  • Verified SPA episode navigation works without WASM reload

vertix added 3 commits March 7, 2026 12:38
- Align `scheduler_decay_steps` with total training steps by default,
  preventing LR from hitting floor early in long runs
- Add `jinja2` to dependencies (required by starlette's Jinja2Templates)
- Remove `full_report=True` from sim eval `StackingSuccess` observer
- Fix SmolVLAConfig import (was XVLAConfig)
- Add explicit `batch_size=64` (SmolVLA recommended default)
- Extract `_train()` to support multiple training configs
- Add `full_finetune` config (unfreezes vision encoder + all params)
- Multi-config CLI: `train` and `full_finetune` subcommands
Use rerun viewer's `add_receiver`/`remove_receiver` API to swap RRD
data without reloading the iframe. Adds Service Worker to cache WASM
assets and `/api/episode/{id}` endpoint for client-side navigation.
@vertix vertix changed the title Fix lerobot-train LR decay schedule and add jinja2 dep SmolVLA training configs and SPA episode viewer Mar 10, 2026
vertix added 2 commits March 10, 2026 08:09
- Reset `viewerReady` on iframe fallback reload
- Use `replaceState` in popstate handler to avoid double history entries
- Restore `full_report=True` in sim config (unintentional removal)
@vertix
Copy link
Copy Markdown
Contributor Author

vertix commented Mar 13, 2026

Split into separate PRs: #345 (rerun SPA navigation) and #346 (SmolVLA training fixes)

@vertix vertix closed this Mar 13, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant