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

feat: start of debug setup, experiment description,..

Browse source
This commit is contained in:
cedric 2026-04-10 10:17:18 +00:00
parent 1c64dc20ae
commit ac352ef431
3 changed files with 108 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
scripts/train.py
View file

@ -68,7 +68,19 @@ if __name__ == "__main__":
print_config(args, title="PPO Training Configuration")
env = make_env(args.env_config_path, args.num_envs)
raw_env = env.raw
print(
"\n\n\n Observation space \n",
raw_env.observation_space,
"Action space \n",
raw_env.action_space,
)
print(
"\n\n\n Observation space \n",
raw_env.observation_space,
"Action space \n",
raw_env.action_space,
)
torch.backends.cudnn.deterministic = args.torch_deterministic
ppo_trainer = PPOTrainer(args, env, run_dir, run_name)