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

feat: more generic naming for critic, since messager will use the same

Browse source
This commit is contained in:
cedric 2026-04-02 18:11:49 +00:00
parent 729f122be4
commit 40c7e32344
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/train.py
View file

@ -19,7 +19,7 @@ from torch.utils.tensorboard import SummaryWriter
from brittle_star_project.dataclasses import PPOArgs
from brittle_star_project.dataclasses.EpisodeStatistics import EpisodeStatistics
from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
from MLPs.mlps import SemiGenericNetwork, Actor, Critic, AgentParams, Storage
from MLPs.mlps import SemiGenericNetwork, Actor, OneDenseLayerMLP, AgentParams, Storage
from ppo import PPO
@ -119,7 +119,8 @@ def train(args: PPOArgs):
network = SemiGenericNetwork()
critic_network = SemiGenericNetwork()
actor = Actor(action_dim=env.single_action_space.shape[0]) # continuous actions for MJX
critic = Critic()
critic = OneDenseLayerMLP()
# messager = OneDenseLayerMLP()
sample_obs = jnp.concatenate(
[