Skip to content

Bernhard 01#536

Open
Kait0 wants to merge 33 commits into
3.0from
bernhard_01
Open

Bernhard 01#536
Kait0 wants to merge 33 commits into
3.0from
bernhard_01

Conversation

@Kait0

@Kait0 Kait0 commented Jul 23, 2026

Copy link
Copy Markdown
Collaborator

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

The pull request also includes some of my cluster files for nodes and job submissions.
Not sure if there is a way to exclude them from the merge. If anybody knows, let me know. Otherwise they won't be in the way.
This is my first pull request. Ideally, somebody familiar with the repo should review it. We can also do a meeting.

@eugenevinitsky

Copy link
Copy Markdown

You can just remove the files from the git commit, no? Is there an issue that I'm not seeing

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