Add README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: pytorch
|
| 3 |
+
tags:
|
| 4 |
+
- autonomous-driving
|
| 5 |
+
- flow-matching
|
| 6 |
+
- occupancy-prediction
|
| 7 |
+
- trajectory-prediction
|
| 8 |
+
- bench2drive
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# OCCWAM B2D v9 step18000
|
| 12 |
+
|
| 13 |
+
This repository contains the deployed OCCWAM joint flow-matching checkpoint
|
| 14 |
+
used by the v9 Bench2Drive closed-loop agent.
|
| 15 |
+
|
| 16 |
+
## Checkpoint
|
| 17 |
+
|
| 18 |
+
| Field | Value |
|
| 19 |
+
| --- | --- |
|
| 20 |
+
| File | `v9_step18000/v9step18000_FULL_merged.pt` |
|
| 21 |
+
| Size | 1,656,918,530 bytes |
|
| 22 |
+
| SHA-256 | `43d914c7afa8c892b03023c96a6f3619b1aa37d874e103d28a12824a1c42316f` |
|
| 23 |
+
| Envelope | `occwam-fullmot-best-model-v1` |
|
| 24 |
+
| Global step | 18000 |
|
| 25 |
+
| Config SHA-256 | `7948cd560c83dc2b519be5ff882c5ff68f43daf3456e4fb774d4ae659017651b` |
|
| 26 |
+
|
| 27 |
+
Training and inference code: <https://github.com/HQH111/OCCWAM>
|
| 28 |
+
|
| 29 |
+
The inference graph is scene-frame joint OCC/trajectory flow matching from
|
| 30 |
+
white noise, followed by a hard scene-to-future-ego warp, VAE re-encoding and
|
| 31 |
+
one velocity-parameterised ego-rollout refinement.
|
| 32 |
+
|
| 33 |
+
## Required contract
|
| 34 |
+
|
| 35 |
+
```bash
|
| 36 |
+
export AEC_TRAJ1_CTX=2
|
| 37 |
+
export AEC_TRAJ2_VFM=1
|
| 38 |
+
export OCCWAM_TP_NORM_UNIFIED=1
|
| 39 |
+
```
|
| 40 |
+
|
| 41 |
+
The unified target-point normalisation flag must be set at both training and
|
| 42 |
+
inference. Omitting it silently changes the input conditioning.
|
| 43 |
+
|
| 44 |
+
This checkpoint does not embed the frozen Qwen3-VL, Stage-3 carrier, OCC VAE,
|
| 45 |
+
TransFuser++ and Profile-B assets. Their hashes and roles are documented in the
|
| 46 |
+
GitHub repository's runtime and training configs.
|
| 47 |
+
|