tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

fix: dubble reshape bad

Browse source
This commit is contained in:
JibrilExe 2026-05-01 10:43:27 +02:00
parent 22c43f05b1
commit e867dbab49
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/brittle_star_project/trainers/PPOTrainer.py
View file

@ -234,7 +234,6 @@ def _get_action_and_value_noise(
logprob = -0.5 * (((raw_action - mean) / std) ** 2 + 2 * log_std + jnp.log(2 * jnp.pi)).sum(-1)
value = apply_shared(critic, agent_state.params["critic_params"], hidden_critic)
raw_action = raw_action.reshape(-1)
raw_action = raw_action.reshape(raw_action.shape[0], -1) # concat the per agent, keep the envs dim
clipped_action = _clip_action(raw_action, action_low, action_high)
return clipped_action, raw_action, logprob, value.squeeze(-1), mean, std, key