Skip to content

Bernhard add mean action - #541

Open
Kait0 wants to merge 45 commits into
3.0from
bernhard_add_mean_action
Open

Bernhard add mean action#541
Kait0 wants to merge 45 commits into
3.0from
bernhard_add_mean_action

Conversation

@Kait0

@Kait0 Kait0 commented Jul 27, 2026

Copy link
Copy Markdown
Collaborator

The same PR as 536 before, but without my personal files:

Added the option to make the policy use either the mean, the mode, or a sample of the distribution instead of the mode during inference.
This required a fundamental design change.
Previously, the action representation was hardcoded in the simulator.

Now there are two options: one for the agent, either continuous or discrete, and one for the environment, either continuous or discrete (to avoid removing features).
The default now is:
env.action_type: continuous
policy.action_type: discrete

The default for eval is kept in mode so that there is no change in behavior with the default configuration.

In local eval eval_simulation: gigaflow
and default parameters, this improved the latest nightly model slightly:
mean: "avg_distance_per_infraction": 16646
mode: "avg_distance_per_infraction": 16127

I verified that these changes have no influence on training performance.

@Kait0 Kait0 mentioned this pull request Jul 27, 2026
Comment thread pufferlib/pufferl.py
Comment thread pufferlib/ocean/torch.py Outdated
Comment thread pufferlib/ocean/torch.py Outdated
Comment thread pufferlib/pytorch.py Outdated
Comment thread pufferlib/config/puffer_drive.yaml Outdated
Comment thread pufferlib/config/puffer_drive.yaml Outdated
Comment thread pufferlib/config/puffer_drive.yaml Outdated
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.

2 participants