Skip to content

Commit d42150f

Browse files
author
henrique
committed
fix bug with dynamic timeout
1 parent 70a656d commit d42150f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/codeevolve/evolution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -755,7 +755,7 @@ def _do_checkpoint(epoch_num: int) -> None:
755755
if use_map_elites:
756756
logger.info(f"sol_db EliteMap: {sol_db.elite_map.map}")
757757

758-
init_pop_size: int = evolve_config.get("init_pop", sol_db.num_alive)
758+
init_pop_size: int = evolve_config.get("init_pop", 0)
759759
gen_init_pop: bool = sol_db.num_alive < init_pop_size
760760

761761
if not gen_init_pop and exploration_scheduler is not None:
@@ -817,7 +817,7 @@ def _do_checkpoint(epoch_num: int) -> None:
817817
if not gen_init_pop and timeout_scheduler is not None:
818818
child_timeout = int(
819819
timeout_scheduler(
820-
epoch=epoch,
820+
epoch=epoch - init_pop_size,
821821
best_fitness=sol_db.programs[sol_db.best_prog_id].fitness,
822822
)
823823
)

0 commit comments

Comments
 (0)