1
Fork 0

Morphology/brittle star 2 arms (#17)

Provide the features to read in Environment (Morphology, arena, ...) config files in the JSON format. The example JSON file contains the config for a brittle star with 2 arms
This commit is contained in:
RobinMeersman 2026-04-01 19:17:42 +02:00 committed by GitHub
parent ecdbe74df4
commit f594cafead
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 113 additions and 100 deletions

View file

@ -7,6 +7,9 @@ class PPOArgs:
source: https://github.com/vwxyzjn/cleanrl/blob/master/cleanrl/ppo_atari_envpool_xla_jax_scan.py
"""
# path to environment config file, if None, use default config
config_path: str | None = None
# the name of this experiment
exp_name: str = "brittle_star_ppo"
@ -41,8 +44,6 @@ class PPOArgs:
hf_entity: str = ""
# ==== Algorithm specific dataclasses ====
# the id of the environment
env_id: str = "" # todo
# total timesteps of the experiments
total_timesteps: int = 10000000
@ -51,7 +52,7 @@ class PPOArgs:
learning_rate: float = 2.5e-4
# the number of parallel game environments
num_envs: int = 16
num_envs: int = 100
# the number of steps to run in each environment per policy rollout
num_steps: int = 128