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

other: backup of current progress

Browse source
This commit is contained in:
Robin Meersman 2026-05-01 18:15:09 +02:00
parent bceb7fe4cf
commit 9f7c73b009
7 changed files with 109 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/train.py
View file

@ -8,6 +8,7 @@ from brittle_star_project.configs.register_configs import register_configs
from brittle_star_project.trainers.PPOTrainer import PPOTrainer
from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
from experiment_logger import init_logger, get_logger
import logging
def make_env(cfg: BrittleStarConfig) -> BrittleStarJaxEnvWrapper:
@ -41,6 +42,7 @@ def main(dict_cfg: DictConfig):
base_dir=os.path.dirname(run_dir),
)
logger = get_logger()
logger.set_level(logging.DEBUG)
logger.info(f"Hydra-initialized run: {run_name}")
logger.info(f"Output directory: {run_dir}")