Views
No views yet
| Directory | Model | Description |
|---|---|---|
joint-cd/ | FastWAM-Joint-CD | Fast one-step joint world-and-action prediction for LIBERO. |
ac-stream/ | Streaming-WAM | Recommended Streaming-WAM checkpoint for efficient LIBERO evaluation. |
1model.pt
2dataset_stats.jsonChunk Time is the average model inference time for one action chunk. Episode Time is the average end-to-end task duration for long- and short-horizon tasks.| Method | LIBERO-10 | Spatial | Goal | Object | Average (%) ↑ | Chunk Time (ms) ↓ | Episode Time (s) ↓ Long / Short |
|---|---|---|---|---|---|---|---|
| FastWAM | 96.20 | 96.20 | 94.20 | 96.20 | 95.70 | 493.0 | 16.31 / 8.25 |
| FastWAM-Joint-CD | 97.20 | 99.60 | 98.60 | 100.00 | 98.85 | 114.2 | 6.89 / 3.74 |
| Streaming-WAM | 96.60 | 98.80 | 97.40 | 100.00 | 98.20 | 41.0 | 5.36 / 3.15 |
1git clone https://github.com/SJTU-DENG-Lab/Streaming-WAM.git
2cd Streaming-WAM
3
4python -m pip install -U uv
5uv sync1git clone https://github.com/Lifelong-Robot-Learning/LIBERO.git third_party/LIBERO
2uv pip install -e third_party/LIBERO --no-deps
3
4uv run huggingface-cli download Wan-AI/Wan2.2-TI2V-5B \
5 --local-dir checkpoints/Wan2.2-TI2V-5B1hf download SJTU-DENG-Lab/Streaming-WAM \
2 --local-dir checkpoints/streamingwam1checkpoints/streamingwam/
2├── joint-cd/
3│ ├── model.pt
4│ └── dataset_stats.json
5└── ac-stream/
6 ├── model.pt
7 └── dataset_stats.json1PYTHON_BIN=.venv/bin/python \
2GPU_IDS=0,1,2,3 \
3BACKBONE_PATH="$PWD/checkpoints/Wan2.2-TI2V-5B" \
4LIBERO_HOME_PATH="$PWD/third_party/LIBERO" \
5CHECKPOINT_PATH="$PWD/checkpoints/streamingwam/ac-stream/model.pt" \
6STATS_PATH="$PWD/checkpoints/streamingwam/ac-stream/dataset_stats.json" \
7 bash examples/libero/scripts/launch_streamingwam_libero_ac_stream_4gpu.sh \
8 --ac-stream-acceleratedlibero_spatial, libero_object, libero_goal, and libero_10. GPU IDs can be changed through GPU_IDS.1python examples/libero/multigpu_rollout.py \
2 --gpus 0,1,2,3 \
3 --suites libero_spatial,libero_object,libero_goal,libero_10 \
4 --num-trials 1 \
5 --config examples/libero/configs/recipes/streamingwam_libero_joint_cd_wan22_5b.yaml \
6 --checkpoint-format fastwam \
7 --checkpoint checkpoints/streamingwam/joint-cd/model.pt \
8 --backbone-path checkpoints/Wan2.2-TI2V-5B \
9 --stats-path checkpoints/streamingwam/joint-cd/dataset_stats.json \
10 --libero-home third_party/LIBERO \
11 --num-steps-wait 30 \
12 --replan-steps 16 \
13 --num-inference-steps 1 \
14 --sampling-method consistency \
15 --fixed-seed \
16 --mujoco-gl egl \
17 --save-video