Skip to content

10 bernhard dev Cosim PR - #597

Merged
Kait0 merged 341 commits into
3.0from
10_bernhard_dev
Oct 5, 2026
Merged

Kait0 merged 341 commits into
3.0from
10_bernhard_dev

Conversation

@Kait0

@Kait0 Kait0 commented Sep 30, 2026 •

Copy link
Copy Markdown
Collaborator

This PR contains everything for the cosim of nuPlan and CARLA as well as related policy changes to push the performance of the policy on these benchmarks.

In particular:

  • Speed limit randomization via speed zones.
    The CARLA maps have a limited range of speed signs (30km/h, 108Km/h) which doesn't cover the whole distribution and leads to issues on nuPlan and Alpasim. I added speed zone which randomize speed limits plus minus 30% to get better coverage.

  • Stop signs
    Added stop sign to ego input and reward. This is needed for CARLA.

  • SPS optimizations. Improved SPS by 25% by using pinned memory

  • Add nuPlan cosim code

  • Add CARLA cosim code

  • Some smaller bug fixes

  • Remove 20% of traffic light groups in training like in GIGAFLOW paper

  • Optimize Map cache to use less RAM

  • Also includes Valentins truncation changes

  • Add k_scaled_0036 weights our best model in self-play so far.

  • Add speed limits to the lane observation to enable the agent to break preemptively for speed limit changes, avoiding penalty.

This PR includes all the changes + improvements of #562 which should be closed after this is merged.

Latest Run:
https://forge.coreweave.com/wandb/emerge_/nightly-multi-long/runs/k_scaled_0044_1000?nw=nwuserbern_jaeger

Current evals:
Self-play: 718k km/infraction (similar to before)
nuPlan reactive: 91.8 CLS (up from 90 CLS)
nuPlan nonreactive: 88.2 CLS (new)
longest6v2: 48 DS (up from 35 DS)

Ready to merge now

Kait0 added 30 commits August 18, 2026 16:48
Also fix bug where lane assignment didn't have a z filter.
    eval.reward_lane_center=0.0075 \
    configurable and change defaults for the next eval.
Comment thread pufferlib/ocean/drive/binding.c Outdated
Comment thread pufferlib/ocean/drive/binding.c Outdated

@eugenevinitsky eugenevinitsky left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Approved, if possible please tag with a wandb run for posterity

@Kait0

Kait0 commented Oct 5, 2026

Copy link
Copy Markdown
Collaborator Author

Already added the wandb run above. thought the link is now coreweave (think wandb got bought)

@Kait0
Kait0 merged commit 847ce80 into 3.0 Oct 5, 2026
12 checks passed
@Kait0 Kait0 mentioned this pull request Oct 5, 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.

3 participants