Repository navigation
Released checkpoints of Reflex (the ones evaluated in the paper). scripts/download_checkpoints.sh fetches and verifies them against assets/checkpoints.sha256.
| File | Stage |
|---|---|
teacher_stage1.pt |
1 — state-based teacher (PPO + AMP) |
bc_init_stage2.pt |
2 — behaviour-cloning initialisation of encoder + decoder |
latent_stage2.pt |
2 — latent controller after joint PPO |
stage3_heatmap_cnn.pt, stage3_feature_net.pt, stage3_latent_weights.npz, visual_encoder_stage3.pt |
3 — egocentric RGB-D visual encoder and its intermediate steps |
The Stage 3 rendered frame set (2.1 GB) is in the separate release stage3-frames-v1 (scripts/download_stage3_frames.sh). See docs/REPRODUCE.md for what each file reproduces.