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

fix: uniform naming of models accross code

Browse source
This commit is contained in:
cedric 2026-04-02 18:37:37 +00:00
parent 79221372df
commit 1c4e40b5fb
3 changed files with 39 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/MLPs/mlps.py
View file

@ -40,10 +40,10 @@ class Actor(nn.Module):
@jax.tree_util.register_dataclass
@dataclass
class AgentParams:
network_params: flax.core.FrozenDict
sensor_params: flax.core.FrozenDict
actor_params: flax.core.FrozenDict
critic_params: flax.core.FrozenDict
critic_network_params: flax.core.FrozenDict
feature_extractor_params: flax.core.FrozenDict
@jax.tree_util.register_dataclass