IMPRINT RMBench Recurrent Checkpoint
Canonical Pi0.5 IMPRINT checkpoint for RMBench using the recurrent gated-delta memory encoder.
- Stage 1: 160,000 updates, W&B
liqhuiha/Robocasa/9bm5uvc7 - Stage 2: 100,000 updates, W&B
liqhuiha/Robocasa/0f4k18rw - memory tokens: 256 x 1024
- policy horizon: 50
- training scan: chunkwise WY
- inference update: incremental recurrent
- Stage 2: memory encoder finetuned; image tower frozen
The release contains the Stage-1 encoder, the complete Stage-2 Orbax parameter checkpoint (including the PaliGemma/Gemma backbone), normalization statistics, portable history configuration, and provenance. The checkpoint is a Gemma model derivative and is distributed subject to the included Gemma Terms of Use.
Use the matching yilin-wu/imprint-rmbench-450 dataset and the evaluation
contract in the IMPRINT GitHub repository. Evaluation requires strict causal
actions, online recent DINOv3 features, EVAL_CONSOLIDATE_ALL=1, BGR policy
inputs, and 50-step prediction/execution chunks.
The matching training store and reproducibility code are publicly available under the IMPRINT release.