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

fix(PPOTrainer.py, PPOArgs.py): added hyperparam config path to cli args, fixed missing arguments error in _step call

Browse source
This commit is contained in:
Robin Meersman 2026-04-06 14:07:46 +02:00
parent ab7fa1584f
commit 8a2f8b14bc
3 changed files with 16 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
experiments/PPOTrainer.py
View file

@ -500,12 +500,14 @@ class PPOTrainer:
iter_bar = tqdm.tqdm(
range(1, self.args.num_iterations + 1),
disable=not sys.stdout.isatty(),
disable=not is_tty,
)
for iteration in iter_bar:
iteration_time_start = time.time()
env_state, next_obs, next_done, loss_info = self._step(env_state, next_obs, next_done)
env_state, next_obs, next_done, loss_info = self._step(
env_state, next_obs, next_done, is_tty=is_tty, iteration=iteration
)
if not is_tty and iteration == 1:
print(f">>> [HPC] First rollout completed: {time.ctime()}", flush=True)